Home
last modified time | relevance | path

Searched refs:device_assignment_size (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/c/
H A Dpjrt_c_api.h330 size_t device_assignment_size; member
333 PJRT_STRUCT_SIZE(PJRT_CompileOptions, device_assignment_size);
H A Dpjrt_c_api_wrapper_impl.cc215 if (c_option->device_assignment_size > 0) { in ConvertCCompileOptionstoCppCompileOptions()
217 c_option->device_assignment_size); in ConvertCCompileOptionstoCppCompileOptions()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.cc215 c_options.device_assignment_size = device_assignment_str->size(); in ConvertCppCompileOptionsToCCompileOptions()
217 c_options.device_assignment_size = 0; in ConvertCppCompileOptionsToCCompileOptions()