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()180 outputs_.emplace(name, std::vector<TensorDescription>()); in ProcessStepStats()181 std::vector<TensorDescription>* outputs = &(output_result.first->second); in ProcessStepStats()
28 TensorDescription tensor = 3;52 TensorDescription tensor = 4;
13 message TensorDescription { message
38 TensorDescription tensor_description = 3;
46 class TensorDescription; variable542 void FillDescription(TensorDescription* description) const;
917 TensorDescription tensor_description; in AllocatedBytes()1197 void Tensor::FillDescription(TensorDescription* description) const { in FillDescription()
1042 TensorDescription* tensor_descr = no->mutable_tensor_description(); in GenerateRunMetadata()