Searched refs:SE_Executable (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
H A D | tpu_executor_c_api.h | 311 SE_Executable** result, TF_Status* status); 316 SE_DeviceMemoryAllocator* allocator, SE_Executable** executables, 326 SE_Executable* executable, SE_ExecutableRunOptions* se_options, 343 TFTPU_CAPI_EXPORT void TpuExecutable_Fingerprint(SE_Executable* executable, 352 SE_Executable* executable, SE_ExecutableSerializationHandle** handle, 374 SE_Executable** executable, 380 TpuExecutable_HloModule(SE_Executable* executable); 382 TFTPU_CAPI_EXPORT void TpuExecutable_Free(SE_Executable*);
|
H A D | tpu_executable.h | 26 TpuExecutable(SE_Executable* se_executable, in TpuExecutable() 65 SE_Executable* se_executable_;
|
H A D | c_api_decl.h | 157 typedef struct SE_Executable SE_Executable; typedef
|
H A D | tpu_executable.cc | 194 SE_Executable* se_executable; in Deserialize()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
H A D | tpu_on_demand_compiler.cc | 91 SE_Executable* result; in RunBackend() 148 SE_Executable** se_executables = new SE_Executable*[module_group->size()]; in Compile()
|