Home
last modified time | relevance | path

Searched refs:TensorDescription (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dstat_summarizer.h106 void Validate(const std::vector<TensorDescription>* outputs,
109 std::map<std::string, std::vector<TensorDescription> > outputs_;
Dstat_summarizer.cc44 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()
/external/tensorflow/tensorflow/core/framework/
Dlog_memory.proto30 TensorDescription tensor = 3;
54 TensorDescription tensor = 4;
Dtensor_description.proto15 message TensorDescription { message
Dstep_stats.proto40 TensorDescription tensor_description = 3;
Dtensor.h48 class TensorDescription; variable
605 void FillDescription(TensorDescription* description) const;
Dtensor.cc1261 void Tensor::FillDescription(TensorDescription* description) const { in FillDescription()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc1218 TensorDescription* tensor_descr = no->mutable_tensor_description(); in GenerateRunMetadata()