Lines Matching refs:TensorSpec
135 const std::vector<TensorSpec> &InputSpecs,
136 function_ref<TensorSpec(size_t)> GetOutputSpecs,
182 const TensorSpec &OutputSpec);
187 const TensorSpec RewardSpec;
229 const TensorSpec &RewardSpec, bool IncludeReward) in LoggerDataImpl()
281 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluatorImpl()
282 function_ref<TensorSpec(size_t)> GetOutputSpecs, size_t OutputSpecsSize, in TFModelEvaluatorImpl()
334 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluator()
335 function_ref<TensorSpec(size_t)> GetOutputSpecs, size_t OutputSpecsSize, in TFModelEvaluator()
344 const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluator()
345 const std::vector<TensorSpec> &OutputSpecs, in TFModelEvaluator()
365 const TF_Output &Output, const TensorSpec &OutputSpec) { in checkReportAndInvalidate()
431 const TensorSpec &RewardSpec, bool IncludeReward) in Logger()