Searched refs:TensorDescription (Results 1 – 8 of 8) sorted by relevance
106 void Validate(const std::vector<TensorDescription>* outputs,109 std::map<std::string, std::vector<TensorDescription> > outputs_;
44 void StatSummarizer::Validate(const std::vector<TensorDescription>* outputs, in Validate()192 outputs_.emplace(name, std::vector<TensorDescription>()); in ProcessStepStats()193 std::vector<TensorDescription>* outputs = &(output_result.first->second); in ProcessStepStats()
30 TensorDescription tensor = 3;54 TensorDescription tensor = 4;
15 message TensorDescription { message
40 TensorDescription tensor_description = 3;
48 class TensorDescription; variable605 void FillDescription(TensorDescription* description) const;
1261 void Tensor::FillDescription(TensorDescription* description) const { in FillDescription()
1218 TensorDescription* tensor_descr = no->mutable_tensor_description(); in GenerateRunMetadata()