Home
last modified time | relevance | path

Searched defs:AllocateTuple (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
H A Dgrpc_tpu_driver.cc357 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anonb00856100111::GrpcTpuDriver
751 std::unique_ptr<BufferHandle> GrpcTpuStream::AllocateTuple( in AllocateTuple() function in tpu_driver::__anonb00856100111::GrpcTpuStream
H A Ddirect_tpu_driver.cc303 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anon439fe69f0111::DirectTpuDriver
H A Drecording_tpu_driver.cc203 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anon852af9610111::RecordingTpuDriver
H A Dpod_tpu_driver.cc393 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anonf243d9070111::PodTpuDriver