Home
last modified time | relevance | path

Searched refs:PJRT_Executable_AddressableDevices (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/c/
H A Dpjrt_c_api.h539 typedef PJRT_Error* PJRT_Executable_AddressableDevices( typedef
795 _PJRT_API_STRUCT_FIELD(PJRT_Executable_AddressableDevices);
H A Dpjrt_c_api_wrapper_impl.h114 PJRT_Error* PJRT_Executable_AddressableDevices(
H A Dpjrt_c_api_wrapper_impl.cc350 PJRT_Error* PJRT_Executable_AddressableDevices( in PJRT_Executable_AddressableDevices() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.cc437 pjrt::LogFatalIfPjrtError(api->PJRT_Executable_AddressableDevices(&args), in InitDevices()