| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| graph_transfer_utils.cc | 61 const RemoteFusedGraphExecuteUtils::TensorShapeMap& tensor_shape_map) {
81 RemoteFusedGraphExecuteUtils::GetTensorShapeType(tensor_shape_map,
103 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; local
105 *original_def, inputs, true /* initialize_by_zero */, &tensor_shape_map);
108 AddOutputTensorShapeTypeByTensorShapeMap(tensor_shape_map, &node_def));
131 tensor_shape_map);
137 RemoteFusedGraphExecuteUtils::GetTensorShapeType(tensor_shape_map,
|
| graph_transfer_utils.h | 52 const RemoteFusedGraphExecuteUtils::TensorShapeMap& tensor_shape_map);
|
| graph_transferer.cc | 215 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; local
217 graph_def, input_node_info_list, true, &tensor_shape_map);
222 TF_CHECK_OK(AddOutputTensorShapeTypeByTensorShapeMap(tensor_shape_map,
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| remote_fused_graph_execute_utils.h | 110 const bool initialize_by_zero, TensorShapeMap* tensor_shape_map);
120 TensorShapeMap* tensor_shape_map);
130 const TensorShapeMap& tensor_shape_map, NodeDef* node_def);
147 TensorShapeMap* tensor_shape_map);
150 const TensorShapeMap& tensor_shape_map, const string& node_name);
153 const TensorShapeMap& tensor_shape_map, const string& node_name,
291 TensorShapeMap* tensor_shape_map);
|
| remote_fused_graph_execute_utils_test.cc | 89 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; local
298 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; local
306 def, inputs, false /* initialize_by_zero */, &tensor_shape_map);
311 ASSERT_EQ(3, tensor_shape_map.size());
312 ASSERT_EQ(1, tensor_shape_map.count(NAME_A));
313 ASSERT_EQ(1, tensor_shape_map.count(NAME_B));
314 ASSERT_EQ(1, tensor_shape_map.count(NAME_A_PLUS_B));
317 RemoteFusedGraphExecuteUtils::GetTensorShapeType(tensor_shape_map,
323 tst = RemoteFusedGraphExecuteUtils::GetTensorShapeType(tensor_shape_map,
342 RemoteFusedGraphExecuteUtils::TensorShapeMap tensor_shape_map; local
[all...] |
| remote_fused_graph_execute_utils.cc | 269 RemoteFusedGraphExecuteUtils::TensorShapeMap* tensor_shape_map) {
270 CHECK(tensor_shape_map != nullptr);
310 EmplaceTensorShapeType(name, tensor, tensor_shape_map);
315 EmplaceTensorShapeType(name, tensor, tensor_shape_map);
338 TensorShapeMap* tensor_shape_map) {
339 CHECK_NE(tensor_shape_map, nullptr);
340 tensor_shape_map->clear();
341 tensor_shape_map->reserve(input_node_info_list.size() +
348 EmplaceTensorShapeType(node_name, tensor, tensor_shape_map);
374 const TensorShapeMap& tensor_shape_map, NodeDef* node_def)
579 TensorShapeMap tensor_shape_map; local
[all...] |
