Searched defs:input_node_info_list (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/hexagon/
H A Dgraph_transferer_test.cc317 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 Dgraph_transferer.cc86 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 Dremote_fused_graph_execute_utils.cc201 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