Searched refs:executor_api_fn (Results 1 – 2 of 2) sorted by relevance
22 static TfTpu_ExecutorApiFn executor_api_fn; in ExecutorApiFn() local23 return &executor_api_fn; in ExecutorApiFn()26 bool IsStreamExecutorEnabled(TfTpu_ExecutorApiFn* executor_api_fn) { in IsStreamExecutorEnabled() argument27 if (!IsInitialized(executor_api_fn)) { in IsStreamExecutorEnabled()31 auto* tpu_platform = executor_api_fn->TpuPlatform_NewFn(); in IsStreamExecutorEnabled()34 executor_api_fn->TpuPlatform_FreeFn(tpu_platform); in IsStreamExecutorEnabled()39 bool IsInitialized(TfTpu_ExecutorApiFn* executor_api_fn) { in IsInitialized() argument43 return executor_api_fn->TpuPlatform_NewFn != nullptr; in IsInitialized()
29 bool IsStreamExecutorEnabled(TfTpu_ExecutorApiFn* executor_api_fn);33 bool IsInitialized(TfTpu_ExecutorApiFn* executor_api_fn);