Home
last modified time | relevance | path

Searched refs:TpuBufferHandle (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dlibtpu.h43 typedef struct TpuBufferHandle { struct
47 } TpuBufferHandle; typedef
151 int32_t inputc, struct TpuBufferHandle** input_buffer_handle,
152 int32_t outputc, struct TpuBufferHandle** output_buffer_handle,
156 typedef struct TpuBufferHandle*(PrototypeTpuDriver_AllocateTuple)(
158 int32_t bufferc, struct TpuBufferHandle** buffer_handle, int32_t eventc,
161 typedef struct TpuBufferHandle*(PrototypeTpuDriver_Allocate)(
165 typedef struct TpuBufferHandle*(PrototypeTpuDriver_AllocateShape)(
175 struct TpuDriver* driver, struct TpuBufferHandle* buffer_handle,
179 struct TpuDriver* driver, const void* src, struct TpuBufferHandle* dst,
[all …]
Dlibtpu_client.c86 struct TpuBufferHandle* buf_a_handle = in main()
90 struct TpuBufferHandle* buf_b_handle = in main()
94 struct TpuBufferHandle* buf_sum_handle = in main()
118 TpuBufferHandle* input_buffer_handle[] = {buf_a_handle, buf_b_handle}; in main()
119 TpuBufferHandle* output_buffer_handle[] = {buf_sum_handle}; in main()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Ddirect_tpu_driver.cc118 ::TpuBufferHandle* handle) in DirectBufferHandle()
131 ::TpuBufferHandle* handle_;
309 ::TpuBufferHandle** childbuf = new ::TpuBufferHandle*[children.size()]; in AllocateTuple()
436 std::vector<::TpuBufferHandle*> inputv; in ExecuteProgram()
442 std::vector<::TpuBufferHandle*> outputv; in ExecuteProgram()