Lines Matching refs:TensorSpec
227 static const TensorSpec DecisionSpec =
228 TensorSpec::createSpec<int64_t>(DecisionName, {1});
364 TensorSpec::createSpec<type>(#name, shape),
388 std::vector<TensorSpec> InputFeatures;
422 static const TensorSpec Reward = TensorSpec::createSpec<float>("reward", {1});
429 TensorSpec::createSpec<type>(std::string("action_") + #name, shape),
461 TensorSpec::createSpec<float>("action_discount", {1}), in DevelopmentModeEvictionAdvisorAnalysis()
462 TensorSpec::createSpec<int32_t>("action_step_type", {1}), in DevelopmentModeEvictionAdvisorAnalysis()
463 TensorSpec::createSpec<float>("action_reward", {1})}; in DevelopmentModeEvictionAdvisorAnalysis()
468 TensorSpec::createSpec<float>("action_discount", {1}), in DevelopmentModeEvictionAdvisorAnalysis()
469 TensorSpec::createSpec<int32_t>("action_step_type", {1}), in DevelopmentModeEvictionAdvisorAnalysis()
470 TensorSpec::createSpec<float>("action_reward", {1})}; in DevelopmentModeEvictionAdvisorAnalysis()
495 std::vector<TensorSpec> InputFeatures;
496 std::vector<TensorSpec> TrainingInputFeatures;
528 std::vector<TensorSpec> LFS = InputFeatures; in doInitialization()