Lines Matching refs:TensorSpec
79 static const TensorSpec DecisionSpec =
80 TensorSpec::createSpec<float>(DecisionName, {1});
113 TensorSpec::createSpec<type>(#name, shape),
115 static const std::vector<TensorSpec> InputFeatures{
164 static const TensorSpec Reward = TensorSpec::createSpec<float>("reward", {1});
167 TensorSpec::createSpec<type>(std::string("action_") + #name, shape),
169 static const std::vector<TensorSpec> TrainingInputFeatures{
171 TensorSpec::createSpec<float>("action_discount", {1}),
172 TensorSpec::createSpec<int32_t>("action_step_type", {1}),
173 TensorSpec::createSpec<float>("action_reward", {1})}};
246 std::vector<TensorSpec> LFS = InputFeatures; in doInitialization()