Home
last modified time | relevance | path

Searched refs:tensor_shape_map (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transfer_utils.cc63 const RemoteFusedGraphExecuteUtils::TensorShapeMap& tensor_shape_map) { in BuildRemoteFusedGraphExecuteInfo() argument
83 RemoteFusedGraphExecuteUtils::GetTensorShapeType(tensor_shape_map, in BuildRemoteFusedGraphExecuteInfo()
105 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in BuildFusedGraphDef() local
107 *original_def, inputs, true /* initialize_by_zero */, &tensor_shape_map); in BuildFusedGraphDef()
110 AddOutputTensorShapeTypeByTensorShapeMap(tensor_shape_map, &node_def)); in BuildFusedGraphDef()
133 tensor_shape_map); in BuildFusedGraphDef()
139 RemoteFusedGraphExecuteUtils::GetTensorShapeType(tensor_shape_map, in BuildFusedGraphDef()
Dgraph_transfer_utils.h52 const RemoteFusedGraphExecuteUtils::TensorShapeMap& tensor_shape_map);
Dgraph_transferer.cc223 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in LoadGraphFromProtoFile() local
225 graph_def, input_node_info_list, true, &tensor_shape_map); in LoadGraphFromProtoFile()
230 TF_CHECK_OK(AddOutputTensorShapeTypeByTensorShapeMap(tensor_shape_map, in LoadGraphFromProtoFile()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils_test.cc90 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in FuseByInOut() local
299 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in TEST() local
307 def, inputs, false /* initialize_by_zero */, &tensor_shape_map); in TEST()
312 ASSERT_EQ(3, tensor_shape_map.size()); in TEST()
313 ASSERT_EQ(1, tensor_shape_map.count(NAME_A)); in TEST()
314 ASSERT_EQ(1, tensor_shape_map.count(NAME_B)); in TEST()
315 ASSERT_EQ(1, tensor_shape_map.count(NAME_A_PLUS_B)); in TEST()
318 RemoteFusedGraphExecuteUtils::GetTensorShapeType(tensor_shape_map, in TEST()
324 tst = RemoteFusedGraphExecuteUtils::GetTensorShapeType(tensor_shape_map, in TEST()
343 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; in TEST() local
[all …]
Dremote_fused_graph_execute_utils.h119 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() argument
272 CHECK(tensor_shape_map != nullptr); in DryRunInferenceForAllNode()
312 EmplaceTensorShapeType(name, tensor, tensor_shape_map); in DryRunInferenceForAllNode()
317 EmplaceTensorShapeType(name, tensor, tensor_shape_map); in DryRunInferenceForAllNode()
340 TensorShapeMap* tensor_shape_map) { in ConvertToTensorShapeMap() argument
341 CHECK_NE(tensor_shape_map, nullptr); in ConvertToTensorShapeMap()
342 tensor_shape_map->clear(); in ConvertToTensorShapeMap()
343 tensor_shape_map->reserve(input_node_info_list.size() + in ConvertToTensorShapeMap()
350 EmplaceTensorShapeType(node_name, tensor, tensor_shape_map); in ConvertToTensorShapeMap()
376 const TensorShapeMap& tensor_shape_map, NodeDef* node_def) { in AddOutputTensorShapeTypeByTensorShapeMap() argument
[all …]