Searched refs:tensor_shape_map (Results 1 - 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/hexagon/ |
H A D | 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, 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 D | graph_transfer_utils.h | 52 const RemoteFusedGraphExecuteUtils::TensorShapeMap& tensor_shape_map);
|
H A D | 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,
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | 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);
|
H A D | 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...] |
H A D | 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, 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