Lines Matching refs:TensorSpec
20 #include "llvm/Analysis/TensorSpec.h"
80 static const TensorSpec DecisionSpec =
81 TensorSpec::createSpec<float>(DecisionName, {1});
114 TensorSpec::createSpec<type>(#name, shape),
116 static const std::vector<TensorSpec> InputFeatures{
165 static const TensorSpec Reward = TensorSpec::createSpec<float>("reward", {1});
168 TensorSpec::createSpec<type>(std::string("action_") + #name, shape),
170 static const std::vector<TensorSpec> TrainingInputFeatures{
172 TensorSpec::createSpec<float>("action_discount", {1}),
173 TensorSpec::createSpec<int32_t>("action_step_type", {1}),
174 TensorSpec::createSpec<float>("action_reward", {1})}};
247 std::vector<TensorSpec> LFS = InputFeatures;