Searched refs:execute_info_ (Results 1 – 3 of 3) sorted by relevance
30 : OpKernel(ctx), execute_info_() { in RemoteFusedGraphExecuteOp()38 execute_info_.ParseFromString(serialized_proto); in RemoteFusedGraphExecuteOp()39 if (!execute_info_.executor_name().empty()) { in RemoteFusedGraphExecuteOp()42 execute_info_.executor_name()); in RemoteFusedGraphExecuteOp()48 << execute_info_.executor_name(); in RemoteFusedGraphExecuteOp()54 remote_fused_graph_executor_->Init(execute_info_); in RemoteFusedGraphExecuteOp()57 execute_info_.clear_serialized_executor_parameters(); in RemoteFusedGraphExecuteOp()77 const int graph_input_count = execute_info_.graph_input_node_name_size(); in Compute()81 << ", gt input count = " << execute_info_.graph_input_node_name_size() in Compute()87 const string& input_node_name = execute_info_.graph_input_node_name(i); in Compute()[all …]
95 execute_info_ = &info; in Init()103 CHECK_NOTNULL(execute_info_); in Init()104 for (int i = 0; i < execute_info_->graph_input_node_name_size(); ++i) { in Init()105 const string& input = execute_info_->graph_input_node_name(i); in Init()109 execute_info_->default_graph_input_tensor_shape(i); in Init()116 for (int i = 0; i < execute_info_->graph_output_node_name_size(); ++i) { in Init()117 const string& output = execute_info_->graph_output_node_name(i); in Init()120 execute_info_->default_graph_output_tensor_shape(i); in Init()356 CHECK_NE(execute_info_, nullptr); in ReadOutputNode()359 for (int i = 0; i < execute_info_->graph_output_node_name_size(); ++i) { in ReadOutputNode()[all …]
73 const RemoteFusedGraphExecuteInfo* execute_info_{};