xla_run_args.push_back(o);
    }
    input_idx++;
  }
  absl::c_copy(cluster_info.resource_inputs, std::back_inserter(xla_run_args));
  return xla_run_args;
}

absl::StatusOr<MemoryTypeVector> GetOutputMemoryTypes(const Scope& root,
                                                      Node* n) {
  MemoryTypeVector input_mtypes, output_mtypes;
  DeviceType device_type("");
  TF_RETURN_IF_ERROR(

      // host memory.
      input_memory_types[i] = tensorflow::HOST_MEMORY;
    }
  }
  return input_memory_types;
}

tensorflow::MemoryTypeVector GetOutputMemoryTypes(
    const tensorflow::FunctionBody* fbody) {
  tensorflow::MemoryTypeVector output_memory_types(fbody->ret_types.size(),
                                                   tensorflow::DEVICE_MEMORY);