Searched refs:initialize_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_api_dlsym_initializer.cc | 55 void (*initialize_fn)(bool init_library, int num_args, const char** args);
56 initialize_fn = reinterpret_cast<decltype(initialize_fn)>(
58 (*initialize_fn)(/*init_library=*/true, args.second.size(),
|
| D | tpu_executor_dlsym_initializer.cc | 52 void (*initialize_fn)(bool init_library, int num_args, const char** args);
53 initialize_fn = reinterpret_cast<decltype(initialize_fn)>(
55 (*initialize_fn)(/*init_library=*/true, args.second.size(),
|
| D | tpu_model_server_initializer.cc | 52 void (*initialize_fn)(bool init_library, int num_args, const char** args);
53 initialize_fn = reinterpret_cast<decltype(initialize_fn)>(
55 (*initialize_fn)(/*init_library=*/true, args.second.size(),
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | libtpu_client.c | 37 PrototypeTpuDriver_Initialize* initialize_fn; in LoadAndInitializeDriver() local
38 *(void**)(&initialize_fn) = dlsym(handle, "TpuDriver_Initialize"); in LoadAndInitializeDriver()
39 initialize_fn(driver_fn, true); in LoadAndInitializeDriver()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ |
| D | direct_tpu_driver.cc | 245 PrototypeTpuDriver_Initialize* initialize_fn; in DirectTpuDriver() local
246 *reinterpret_cast<void**>(&initialize_fn) = in DirectTpuDriver()
248 initialize_fn(&driver_fn_, /*initialize=*/true); in DirectTpuDriver()
|