Searched refs:execute_info (Results 1 – 5 of 5) sorted by relevance
64 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteInfo() local65 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() local156 StringPiece(execute_info.SerializeAsString())); in BuildFusedGraphDef()
212 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo() local213 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() local251 hexagon_control_wrapper.Init(execute_info); in RunInferenceByHexagonControlWrapper()
105 const RemoteFusedGraphExecuteInfo& execute_info, Scope* root) { in BuildRemoteFusedGraphExecuteOp() argument120 StringPiece(execute_info.SerializeAsString())); in BuildRemoteFusedGraphExecuteOp()130 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteInfo() local131 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 …]
610 const bool require_shape_type, RemoteFusedGraphExecuteInfo* execute_info, in BuildRemoteFusedGraphExecuteInfo() argument612 CHECK_NOTNULL(execute_info); in BuildRemoteFusedGraphExecuteInfo()616 execute_info->Clear(); in BuildRemoteFusedGraphExecuteInfo()617 execute_info->set_executor_name(executor_name); in BuildRemoteFusedGraphExecuteInfo()620 *execute_info->mutable_remote_graph() = subgraph_def; in BuildRemoteFusedGraphExecuteInfo()628 execute_info->add_graph_input_node_name(input); in BuildRemoteFusedGraphExecuteInfo()631 *execute_info->add_default_graph_input_tensor_shape(); in BuildRemoteFusedGraphExecuteInfo()652 execute_info->add_graph_output_node_name(output); in BuildRemoteFusedGraphExecuteInfo()656 *execute_info->add_default_graph_output_tensor_shape(); in BuildRemoteFusedGraphExecuteInfo()684 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteOpNode() local[all …]
178 const bool require_shape_type, RemoteFusedGraphExecuteInfo* execute_info,