Home
last modified time | relevance | path

Searched refs:execute_info (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transfer_utils.cc64 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteInfo() local
65 execute_info.set_executor_name("build_hexagon_remote_fused_graph_executor"); in BuildRemoteFusedGraphExecuteInfo()
68 *execute_info.mutable_remote_graph() = graph_def; in BuildRemoteFusedGraphExecuteInfo()
71 execute_info.add_graph_input_node_name(input.first); in BuildRemoteFusedGraphExecuteInfo()
73 *execute_info.add_default_graph_input_tensor_shape(); in BuildRemoteFusedGraphExecuteInfo()
86 execute_info.add_graph_output_node_name(output_name); in BuildRemoteFusedGraphExecuteInfo()
88 *execute_info.add_default_graph_output_tensor_shape(); in BuildRemoteFusedGraphExecuteInfo()
97 return execute_info; in BuildRemoteFusedGraphExecuteInfo()
131 const RemoteFusedGraphExecuteInfo execute_info = in BuildFusedGraphDef() local
156 StringPiece(execute_info.SerializeAsString())); in BuildFusedGraphDef()
Dhexagon_graph_execution_test.cc212 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo() local
213 execute_info.set_executor_name("build_hexagon_remote_fused_graph_executor"); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
216 execute_info.add_graph_input_node_name(input.name()); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
218 *execute_info.add_default_graph_input_tensor_shape(); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
228 execute_info.add_graph_output_node_name(output.name()); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
230 *execute_info.add_default_graph_output_tensor_shape(); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
238 execute_info.set_serialized_executor_parameters( in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
240 return execute_info; in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
245 const RemoteFusedGraphExecuteInfo execute_info = in RunInferenceByHexagonControlWrapper() local
251 hexagon_control_wrapper.Init(execute_info); in RunInferenceByHexagonControlWrapper()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test.cc105 const RemoteFusedGraphExecuteInfo& execute_info, Scope* root) { in BuildRemoteFusedGraphExecuteOp() argument
120 StringPiece(execute_info.SerializeAsString())); in BuildRemoteFusedGraphExecuteOp()
130 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteInfo() local
131 execute_info.set_executor_name(REMOTE_FUSED_EXECUTOR_NAME); in BuildRemoteFusedGraphExecuteInfo()
136 NodeDef& copied_node = *execute_info.mutable_remote_graph()->add_node(); in BuildRemoteFusedGraphExecuteInfo()
146 execute_info.add_graph_input_node_name(NAME_A); in BuildRemoteFusedGraphExecuteInfo()
148 *execute_info.add_default_graph_input_tensor_shape(); in BuildRemoteFusedGraphExecuteInfo()
153 execute_info.add_graph_output_node_name(NAME_A_PLUS_B); in BuildRemoteFusedGraphExecuteInfo()
155 *execute_info.add_default_graph_output_tensor_shape(); in BuildRemoteFusedGraphExecuteInfo()
159 return execute_info; in BuildRemoteFusedGraphExecuteInfo()
[all …]
Dremote_fused_graph_execute_utils.cc609 const bool require_shape_type, RemoteFusedGraphExecuteInfo* execute_info, in BuildRemoteFusedGraphExecuteInfo() argument
611 CHECK_NOTNULL(execute_info); in BuildRemoteFusedGraphExecuteInfo()
615 execute_info->Clear(); in BuildRemoteFusedGraphExecuteInfo()
616 execute_info->set_executor_name(executor_name); in BuildRemoteFusedGraphExecuteInfo()
619 *execute_info->mutable_remote_graph() = subgraph_def; in BuildRemoteFusedGraphExecuteInfo()
627 execute_info->add_graph_input_node_name(input); in BuildRemoteFusedGraphExecuteInfo()
630 *execute_info->add_default_graph_input_tensor_shape(); in BuildRemoteFusedGraphExecuteInfo()
651 execute_info->add_graph_output_node_name(output); in BuildRemoteFusedGraphExecuteInfo()
655 *execute_info->add_default_graph_output_tensor_shape(); in BuildRemoteFusedGraphExecuteInfo()
683 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteOpNode() local
[all …]
Dremote_fused_graph_execute_utils.h178 const bool require_shape_type, RemoteFusedGraphExecuteInfo* execute_info,