Lines Matching refs:TensorSpec
18 #include "llvm/Analysis/TensorSpec.h"
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}),
462 TensorSpec::createSpec<int32_t>("action_step_type", {1}),
463 TensorSpec::createSpec<float>("action_reward", {1})};
468 TensorSpec::createSpec<float>("action_discount", {1}),
469 TensorSpec::createSpec<int32_t>("action_step_type", {1}),
470 TensorSpec::createSpec<float>("action_reward", {1})};
495 std::vector<TensorSpec> InputFeatures;
496 std::vector<TensorSpec> TrainingInputFeatures;
528 std::vector<TensorSpec> LFS = InputFeatures;