Home
last modified time | relevance | path

Searched refs:tensor_properties (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc2269 auto tensor_properties = GetOutputProperties(node->name()); in AnnotateOutputShapes() local
2270 for (const auto& tensor_property : tensor_properties) { in AnnotateOutputShapes()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc918 const auto& tensor_properties = output_params.at(output_port); in ConvertToTensorOrWeights() local
919 const DataType dtype = tensor_properties.dtype(); in ConvertToTensorOrWeights()
920 const PartialTensorShape shape = tensor_properties.shape(); in ConvertToTensorOrWeights()