Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/
Dtpu_ops_c_api.h94 TFTPU_CAPI_EXPORT void TpuCompile_CompileAndBuild(
104 TFTPU_CAPI_EXPORT void TpuCompile_XrtCompileAndBuild(
109 TFTPU_CAPI_EXPORT void TpuProfiler_Create(TpuProfiler** tpu_profiler,
112 TFTPU_CAPI_EXPORT void TpuProfiler_Destroy(TpuProfiler* tpu_profiler);
114 TFTPU_CAPI_EXPORT void TpuProfiler_Start(TpuProfiler* tpu_profiler,
117 TFTPU_CAPI_EXPORT void TpuProfiler_Stop(TpuProfiler* tpu_profiler,
138 TFTPU_CAPI_EXPORT void TpuProfiler_CollectData(TpuProfiler* tpu_profiler,
144 TFTPU_CAPI_EXPORT XLA_TpuMeshState* TpuMeshState_Create();
148 TFTPU_CAPI_EXPORT void TpuMeshState_Free(XLA_TpuMeshState* mesh_state);
151 TFTPU_CAPI_EXPORT void* TpuMeshState_MeshCommonState(
[all …]
Dlibtftpu.h27 #define TFTPU_CAPI_EXPORT macro
31 #define TFTPU_CAPI_EXPORT __declspec(dllexport) macro
33 #define TFTPU_CAPI_EXPORT __declspec(dllimport) macro
36 #define TFTPU_CAPI_EXPORT __attribute__((visibility("default"))) macro
44 TFTPU_CAPI_EXPORT void TfTpu_Initialize(bool init_library, int num_args,
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor_c_api.h285 TFTPU_CAPI_EXPORT Tpu_Compiler* TpuCompiler_New();
286 TFTPU_CAPI_EXPORT void TpuCompiler_Free(Tpu_Compiler* compiler);
288 TFTPU_CAPI_EXPORT void TpuCompiler_RunHloPasses(
293 TFTPU_CAPI_EXPORT void TpuCompiler_RunBackend(
298 TFTPU_CAPI_EXPORT void TpuCompiler_Compile(
304 TFTPU_CAPI_EXPORT int64_t TpuCompiler_ShapeSize(Tpu_Compiler* compiler,
307 TFTPU_CAPI_EXPORT void TpuExecutable_ExecuteAsyncOnStream(
315 TFTPU_CAPI_EXPORT void TpuExecutable_FreeXlaShapeIndexArray(
322 TFTPU_CAPI_EXPORT void TpuExecutable_FreeMaybeOwningDeviceMemoryArray(
325 TFTPU_CAPI_EXPORT void TpuExecutable_Fingerprint(SE_Executable* executable,
[all …]