Home
last modified time | relevance | path

Searched refs:TpuDriver_AllocateShape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dlibtpu.h241 TpuDriver_AllocateShape;
284 PrototypeTpuDriver_AllocateShape* TpuDriver_AllocateShape; // NOLINT member
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Ddirect_tpu_driver.cc295 driver_fn_.TpuDriver_AllocateShape(driver_, core_id, region, shape_, in Allocate()