Home
last modified time | relevance | path

Searched refs:SE_DeviceMemoryBase (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.h39 SE_DeviceMemoryBase ToC(const stream_executor::DeviceMemoryBase& base);
41 SE_DeviceMemoryBase* se_base);
42 stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base);
43 void Free(SE_DeviceMemoryBase*);
76 SE_DeviceMemoryBase ToC(const stream_executor::DeviceMemoryBase& base);
77 stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base);
78 void Free(SE_DeviceMemoryBase*);
Dtpu_executable.cc42 SE_DeviceMemoryBase* arguments_bases = nullptr; in LoadProgramAndEnqueueToStream()
44 arguments_bases = new SE_DeviceMemoryBase[arguments.size()]; in LoadProgramAndEnqueueToStream()
47 SE_DeviceMemoryBase{const_cast<void*>(arguments[i].opaque()), in LoadProgramAndEnqueueToStream()
52 SE_DeviceMemoryBase result_base{result.opaque(), result.size(), in LoadProgramAndEnqueueToStream()
54 SE_DeviceMemoryBase prefetch_base; in LoadProgramAndEnqueueToStream()
56 prefetch_base = SE_DeviceMemoryBase{cross_program_prefetch_addr->opaque(), in LoadProgramAndEnqueueToStream()
Dc_api_decl.h71 typedef struct SE_DeviceMemoryBase { struct
75 } SE_DeviceMemoryBase; argument
78 SE_DeviceMemoryBase wrapped;
107 typedef void (*SE_DeallocateFn)(void* ctx, SE_DeviceMemoryBase* base,
174 SE_DeviceMemoryBase memory;
233 SE_DeviceMemoryBase* bases;
Dtpu_executor_c_api.h51 SE_DeviceMemoryBase TpuExecutor_Allocate(SE_StreamExecutor* executor,
54 SE_DeviceMemoryBase* memory);
90 const SE_DeviceMemoryBase* device_src,
93 SE_DeviceMemoryBase* device_dst,
98 const SE_DeviceMemoryBase* device_src,
102 SE_DeviceMemoryBase* device_dst,
131 SE_DeviceMemoryBase device_dst,
135 SE_DeviceMemoryBase device_src,
139 SE_DeviceMemoryBase send_buffer,
140 SE_DeviceMemoryBase recv_buffer,
[all …]
Dtpu_executor.cc196 SE_DeviceMemoryBase se_base = tpu::ExecutorApiFn()->TpuExecutor_AllocateFn( in Allocate()
202 SE_DeviceMemoryBase se_base = ApiConverter::ToC(memory); in Deallocate()
207 SE_DeviceMemoryBase se_base = ApiConverter::ToC(*memory); in Deallocate()
282 SE_DeviceMemoryBase se_base = ApiConverter::ToC(device_src); in Memcpy()
291 SE_DeviceMemoryBase se_base = ApiConverter::ToC(*device_dst); in Memcpy()
301 SE_DeviceMemoryBase se_base = ApiConverter::ToC(*device_dst); in SynchronousMemcpy()
311 SE_DeviceMemoryBase se_base = ApiConverter::ToC(device_src); in SynchronousMemcpy()
Dc_api_conversions.cc109 se_allocator.deallocate = [](void* ctx, SE_DeviceMemoryBase* base, in ToC()
132 SE_DeviceMemoryBase* se_base) { in ToC()
138 SE_DeviceMemoryBase ToC(const stream_executor::DeviceMemoryBase& base) { in ToC()
139 SE_DeviceMemoryBase se_base; in ToC()
144 stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base) { in FromC()
355 absl::InlinedVector<SE_DeviceMemoryBase, 2> bases; in ToC()
360 c_device_buffer->bases = new SE_DeviceMemoryBase[bases.size()]; in ToC()
367 void Free(SE_DeviceMemoryBase*) {} in Free() argument
Dtpu_transfer_manager.cc260 SE_DeviceMemoryBase c_device_buffer{const_cast<void*>(device_buffer.opaque()), in CanBufferBeAccessedNow()
272 SE_DeviceMemoryBase* elements_bases = in WriteSingleTupleIndexTable()
273 new SE_DeviceMemoryBase[elements.size()]; in WriteSingleTupleIndexTable()
276 SE_DeviceMemoryBase{const_cast<void*>(elements[i].opaque()), in WriteSingleTupleIndexTable()
281 SE_DeviceMemoryBase region_base{region->opaque(), region->size(), in WriteSingleTupleIndexTable()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_ops_c_api.h159 SE_DeviceMemoryBase* arguments;
161 SE_DeviceMemoryBase* result;
163 SE_DeviceMemoryBase* cross_program_prefetch_addr;