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

/external/tensorflow/tensorflow/core/kernels/hexagon/
H A Dgraph_transfer_utils.cc61 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,
57 BuildRemoteFusedGraphExecuteInfo( const GraphDef& graph_def, const std::vector<std::pair<string, Tensor>>& inputs, const std::vector<string>& outputs, const RemoteFusedGraphExecuteUtils::TensorShapeMap& tensor_shape_map) argument
H A Dgraph_transfer_utils.h52 const RemoteFusedGraphExecuteUtils::TensorShapeMap& tensor_shape_map);
H A Dgraph_transferer.cc215 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,
/external/tensorflow/tensorflow/core/kernels/
H A Dremote_fused_graph_execute_utils.h110 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);
H A Dremote_fused_graph_execute_utils_test.cc89 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...]
H A Dremote_fused_graph_execute_utils.cc269 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, NodeDe
265 DryRunInferenceForAllNode( const GraphDef& graph_def, const std::vector<std::pair<string, Tensor>>& input_node_info_list, const bool initialize_by_zero, RemoteFusedGraphExecuteUtils::TensorShapeMap* tensor_shape_map) argument
334 ConvertToTensorShapeMap( const std::vector<std::pair<string, Tensor>>& input_node_info_list, const std::vector<string>& output_node_names, const std::vector<tensorflow::Tensor>& output_tensors, TensorShapeMap* tensor_shape_map) argument
373 AddOutputTensorShapeTypeByTensorShapeMap( const TensorShapeMap& tensor_shape_map, NodeDef* node_def) argument
486 BuildTensorShapeMapFromGraph( const Graph& graph, const ShapeRefiner& shape_refiner, TensorShapeMap* tensor_shape_map) argument
519 GetTensorShapeType( const TensorShapeMap& tensor_shape_map, const string& node_name) argument
531 GetTensorShapeType( const TensorShapeMap& tensor_shape_map, const string& node_name, const int port) argument
565 EmplaceTensorShapeType( const string& name, const Tensor& tensor, TensorShapeMap* tensor_shape_map) argument
579 TensorShapeMap tensor_shape_map; local
[all...]

Completed in 175 milliseconds