Searched refs:TensorShapeMap (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | remote_fused_graph_execute_utils.h | 91 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);
|
D | remote_fused_graph_execute_utils.cc | 271 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()
|
D | remote_fused_graph_execute_utils_test.cc | 298 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in TEST() 342 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in TEST()
|
/external/tensorflow/tensorflow/core/kernels/hexagon/ |
D | graph_transferer.h | 50 using TensorShapeMap = RemoteFusedGraphExecuteUtils::TensorShapeMap; variable
|
D | graph_transfer_utils.cc | 63 const RemoteFusedGraphExecuteUtils::TensorShapeMap& tensor_shape_map) { in BuildRemoteFusedGraphExecuteInfo() 105 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in BuildFusedGraphDef()
|
D | graph_transfer_utils.h | 52 const RemoteFusedGraphExecuteUtils::TensorShapeMap& tensor_shape_map);
|
D | graph_transferer_test.cc | 50 const RemoteFusedGraphExecuteUtils::TensorShapeMap EMPTY_OUTPUT_TENSOR_MAP; 303 RemoteFusedGraphExecuteUtils::TensorShapeMap output_tensor_info; in TEST_F()
|
D | graph_transferer.cc | 223 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in LoadGraphFromProtoFile()
|