| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.h | 62 const std::vector<std::pair<string, Tensor>>& input_node_info_list,
72 const std::vector<std::pair<string, Tensor>>& input_node_info_list,
115 const std::vector<std::pair<string, Tensor>>& input_node_info_list,
121 const std::vector<std::pair<string, Tensor>>& input_node_info_list,
172 const std::vector<std::pair<string, Tensor>>& input_node_info_list,
|
| D | graph_transferer_test.cc | 315 std::vector<std::pair<string, Tensor>> input_node_info_list; in TEST_F() local
316 input_node_info_list.emplace_back( in TEST_F()
320 input_node_info_list, output_node_names, in TEST_F()
341 std::vector<std::pair<string, Tensor>> input_node_info_list; in TEST_F() local
342 input_node_info_list.emplace_back( in TEST_F()
346 input_node_info_list, output_node_names, in TEST_F()
378 std::vector<std::pair<string, Tensor>> input_node_info_list = {}; in TEST() local
393 *ops_definitions, filename, input_node_info_list, output_node_names, in TEST()
441 std::vector<std::pair<string, Tensor>> input_node_info_list = {}; in TEST() local
459 *ops_definitions, filename, input_node_info_list, output_node_names, in TEST()
[all …]
|
| D | graph_transferer.cc | 94 const std::vector<std::pair<string, Tensor>>& input_node_info_list, in LoadGraphFromProto() argument
106 graph_def, input_node_info_list, &graph, &shape_refiner); in LoadGraphFromProto()
113 input_node_info_list, &graph, &shape_refiner)); in LoadGraphFromProto()
141 ops_definitions, shape_refiner, *node, false, input_node_info_list, in LoadGraphFromProto()
152 input_node_info_list) { in LoadGraphFromProto()
201 const std::vector<std::pair<string, Tensor>>& input_node_info_list, in LoadGraphFromProtoFile() argument
225 graph_def, input_node_info_list, true, &tensor_shape_map); in LoadGraphFromProtoFile()
235 return LoadGraphFromProto(ops_definitions, graph_def, input_node_info_list, in LoadGraphFromProtoFile()
324 const std::vector<std::pair<string, Tensor>>& input_node_info_list, in TransformGraphToAddAggregatedInputNode() argument
332 for (int i = 0; i < input_node_info_list.size(); ++i) { in TransformGraphToAddAggregatedInputNode()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.h | 107 const std::vector<std::pair<string, Tensor>>& input_node_info_list,
118 const std::vector<std::pair<string, Tensor>>& input_node_info_list,
122 const std::vector<std::pair<string, Tensor>>& input_node_info_list,
126 const std::vector<std::pair<string, Tensor>>& input_node_info_list,
151 const std::vector<std::pair<string, Tensor>>& input_node_info_list,
|
| D | remote_fused_graph_execute_utils.cc | 203 const std::vector<std::pair<string, Tensor>>& input_node_info_list, in DryRunInference() argument
209 for (const std::pair<string, Tensor>& input : input_node_info_list) { in DryRunInference()
269 const std::vector<std::pair<string, Tensor>>& input_node_info_list, in DryRunInferenceForAllNode() argument
284 if (IsInputNode(input_node_info_list, node->name())) { in DryRunInferenceForAllNode()
292 status = DryRunInference(graph_def, input_node_info_list, output_node_names, in DryRunInferenceForAllNode()
305 input_node_info_list) { in DryRunInferenceForAllNode()
314 for (int i = 0; static_cast<size_t>(i) < input_node_info_list.size(); ++i) { in DryRunInferenceForAllNode()
315 const string& name = input_node_info_list.at(i).first; in DryRunInferenceForAllNode()
319 CHECK_EQ(output_node_names.size() + input_node_info_list.size(), in DryRunInferenceForAllNode()
337 const std::vector<std::pair<string, Tensor>>& input_node_info_list, in ConvertToTensorShapeMap() argument
[all …]
|