Searched defs:AllocateTuple (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ | ||
D | grpc_tpu_driver.cc | 357 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver |
D | direct_tpu_driver.cc | 303 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anon9d833a900111::DirectTpuDriver |
D | recording_tpu_driver.cc | 204 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anone63593b20111::RecordingTpuDriver |
D | pod_tpu_driver.cc | 395 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anon6b014a980111::PodTpuDriver |