Searched defs:input_node_info_list (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/hexagon/ |
H A D | graph_transferer_test.cc | 317 std::vector<std::pair<string, Tensor>> input_node_info_list; local 318 input_node_info_list.emplace_back( 322 input_node_info_list, output_node_names, 343 std::vector<std::pair<string, Tensor>> input_node_info_list; local 344 input_node_info_list.emplace_back( 348 input_node_info_list, output_node_names, 381 std::vector<std::pair<string, Tensor>> input_node_info_list = {}; local 387 // input_node_info_list.emplace_back( 396 *ops_definitions, filename, input_node_info_list, output_node_names, 444 std::vector<std::pair<string, Tensor>> input_node_info_list local [all...] |
H A D | graph_transferer.cc | 86 const std::vector<std::pair<string, Tensor>>& input_node_info_list, 98 graph_def, input_node_info_list, &graph, &shape_refiner); 105 input_node_info_list, &graph, &shape_refiner)); 133 ops_definitions, shape_refiner, *node, false, input_node_info_list, 144 input_node_info_list) { 193 const std::vector<std::pair<string, Tensor>>& input_node_info_list, 217 graph_def, input_node_info_list, true, &tensor_shape_map); 227 return LoadGraphFromProto(ops_definitions, graph_def, input_node_info_list, 316 const std::vector<std::pair<string, Tensor>>& input_node_info_list, 324 for (int i = 0; i < input_node_info_list 83 LoadGraphFromProto( const IRemoteFusedGraphOpsDefinitions& ops_definitions, const GraphDef& graph_def, const std::vector<std::pair<string, Tensor>>& input_node_info_list, const std::vector<string>& output_node_names, const bool shape_inference_for_unknown_shape) argument 190 LoadGraphFromProtoFile( const IRemoteFusedGraphOpsDefinitions& ops_definitions, const string& graph_def_path, const std::vector<std::pair<string, Tensor>>& input_node_info_list, const std::vector<string>& output_node_names, const bool is_text_proto, const bool shape_inference_for_unknown_shape, const bool dry_run_for_unknown_shape) argument 315 TransformGraphToAddAggregatedInputNode( const std::vector<std::pair<string, Tensor>>& input_node_info_list, Graph* graph, ShapeRefiner* shape_refiner) argument 414 RegisterNode( const IRemoteFusedGraphOpsDefinitions& ops_definitions, const ShapeRefiner& shape_refiner, const Node& node, const std::vector<std::pair<string, Tensor>>& input_node_info_list, const std::vector<string>& output_node_names) argument 832 RegisterNodeIfAllInputsAreCached( const IRemoteFusedGraphOpsDefinitions& ops_definitions, const ShapeRefiner& shape_refiner, const Node& node, const bool only_register_const_node, const std::vector<std::pair<string, Tensor>>& input_node_info_list, const std::vector<string>& output_node_names) argument [all...] |
/external/tensorflow/tensorflow/core/kernels/ |
H A D | remote_fused_graph_execute_utils.cc | 201 const std::vector<std::pair<string, Tensor>>& input_node_info_list, 207 for (const std::pair<string, Tensor>& input : input_node_info_list) { 267 const std::vector<std::pair<string, Tensor>>& input_node_info_list, 282 if (IsInputNode(input_node_info_list, node->name())) { 290 status = DryRunInference(graph_def, input_node_info_list, output_node_names, 303 input_node_info_list) { 312 for (int i = 0; static_cast<size_t>(i) < input_node_info_list.size(); ++i) { 313 const string& name = input_node_info_list.at(i).first; 317 CHECK_EQ(output_node_names.size() + input_node_info_list.size(), 335 const std::vector<std::pair<string, Tensor>>& input_node_info_list, 199 DryRunInference( const GraphDef& graph_def, const std::vector<std::pair<string, Tensor>>& input_node_info_list, const std::vector<string>& output_node_names, const bool initialize_by_zero, std::vector<tensorflow::Tensor>* output_tensors) argument 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 438 PropagateShapeInference( const GraphDef& graph_def, const std::vector<std::pair<string, Tensor>>& input_node_info_list, Graph* graph, ShapeRefiner* shape_refiner) argument [all...] |
Completed in 132 milliseconds