Home
last modified time | relevance | path

Searched refs:TensorShapeMap (Results 1 – 9 of 9) 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()
490 TensorShapeMap* tensor_shape_map) { in BuildTensorShapeMapFromGraph()
522 const TensorShapeMap& tensor_shape_map, const string& node_name) { in GetTensorShapeType()
533 const TensorShapeMap& tensor_shape_map, const string& node_name, in GetTensorShapeType()
568 TensorShapeMap* tensor_shape_map) { in EmplaceTensorShapeType()
580 TensorShapeMap tensor_shape_map; in BuildAndAddTensorShapes()
Dremote_fused_graph_execute_utils_test.cc90 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in FuseByInOut()
299 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in TEST()
343 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()
Dhexagon_graph_execution_test.cc551 RemoteFusedGraphExecuteUtils::TensorShapeMap output_tensor_info0; in TEST()
571 RemoteFusedGraphExecuteUtils::TensorShapeMap output_tensor_info1; in TEST()
Dgraph_transferer.cc223 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in LoadGraphFromProtoFile()