Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.h91 using TensorShapeMap = std::unordered_multimap<string, // node name variable
119 const bool initialize_by_zero, TensorShapeMap* tensor_shape_map);
129 TensorShapeMap* tensor_shape_map);
139 const TensorShapeMap& tensor_shape_map, NodeDef* node_def);
156 TensorShapeMap* tensor_shape_map);
159 const TensorShapeMap& tensor_shape_map, const string& node_name);
162 const TensorShapeMap& tensor_shape_map, const string& node_name,
300 TensorShapeMap* tensor_shape_map);
Dremote_fused_graph_execute_utils.cc271 RemoteFusedGraphExecuteUtils::TensorShapeMap* tensor_shape_map) { in DryRunInferenceForAllNode()
340 TensorShapeMap* tensor_shape_map) { in ConvertToTensorShapeMap()
376 const TensorShapeMap& tensor_shape_map, NodeDef* node_def) { in AddOutputTensorShapeTypeByTensorShapeMap()
491 TensorShapeMap* tensor_shape_map) { in BuildTensorShapeMapFromGraph()
523 const TensorShapeMap& tensor_shape_map, const string& node_name) { in GetTensorShapeType()
534 const TensorShapeMap& tensor_shape_map, const string& node_name, in GetTensorShapeType()
569 TensorShapeMap* tensor_shape_map) { in EmplaceTensorShapeType()
581 TensorShapeMap tensor_shape_map; in BuildAndAddTensorShapes()
Dremote_fused_graph_execute_utils_test.cc298 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in TEST()
342 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in TEST()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.h50 using TensorShapeMap = RemoteFusedGraphExecuteUtils::TensorShapeMap; variable
Dgraph_transfer_utils.cc63 const RemoteFusedGraphExecuteUtils::TensorShapeMap& tensor_shape_map) { in BuildRemoteFusedGraphExecuteInfo()
105 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in BuildFusedGraphDef()
Dgraph_transfer_utils.h52 const RemoteFusedGraphExecuteUtils::TensorShapeMap& tensor_shape_map);
Dgraph_transferer_test.cc50 const RemoteFusedGraphExecuteUtils::TensorShapeMap EMPTY_OUTPUT_TENSOR_MAP;
303 RemoteFusedGraphExecuteUtils::TensorShapeMap output_tensor_info; in TEST_F()
Dgraph_transferer.cc223 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in LoadGraphFromProtoFile()