Lines Matching refs:TensorSpec
226 static const TensorSpec DecisionSpec =
227 TensorSpec::createSpec<int64_t>(DecisionName, {1});
363 TensorSpec::createSpec<type>(#name, shape),
387 std::vector<TensorSpec> InputFeatures;
420 static const TensorSpec Reward = TensorSpec::createSpec<float>("reward", {1});
427 TensorSpec::createSpec<type>(std::string("action_") + #name, shape),
459 TensorSpec::createSpec<float>("action_discount", {1}), in DevelopmentModeEvictionAdvisorAnalysis()
460 TensorSpec::createSpec<int32_t>("action_step_type", {1}), in DevelopmentModeEvictionAdvisorAnalysis()
461 TensorSpec::createSpec<float>("action_reward", {1})}; in DevelopmentModeEvictionAdvisorAnalysis()
466 TensorSpec::createSpec<float>("action_discount", {1}), in DevelopmentModeEvictionAdvisorAnalysis()
467 TensorSpec::createSpec<int32_t>("action_step_type", {1}), in DevelopmentModeEvictionAdvisorAnalysis()
468 TensorSpec::createSpec<float>("action_reward", {1})}; in DevelopmentModeEvictionAdvisorAnalysis()
493 std::vector<TensorSpec> InputFeatures;
494 std::vector<TensorSpec> TrainingInputFeatures;
526 std::vector<TensorSpec> LFS = InputFeatures; in doInitialization()