Home
last modified time | relevance | path

Searched refs:se_base (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.cc196 SE_DeviceMemoryBase se_base = tpu::ExecutorApiFn()->TpuExecutor_AllocateFn( in Allocate() local
198 return ApiConverter::FromC(se_base); in Allocate()
202 SE_DeviceMemoryBase se_base = ApiConverter::ToC(memory); in Deallocate() local
203 tpu::ExecutorApiFn()->TpuExecutor_DeallocateFn(executor_, &se_base); in Deallocate()
207 SE_DeviceMemoryBase se_base = ApiConverter::ToC(*memory); in Deallocate() local
208 tpu::ExecutorApiFn()->TpuExecutor_DeallocateFn(executor_, &se_base); in Deallocate()
282 SE_DeviceMemoryBase se_base = ApiConverter::ToC(device_src); in Memcpy() local
284 executor_, get_stream(stream->implementation()), host_dst, &se_base, in Memcpy()
291 SE_DeviceMemoryBase se_base = ApiConverter::ToC(*device_dst); in Memcpy() local
293 executor_, get_stream(stream->implementation()), &se_base, host_src, in Memcpy()
[all …]
Dc_api_conversions.cc132 SE_DeviceMemoryBase* se_base) { in ToC() argument
133 se_base->opaque = const_cast<void*>(base.opaque()); in ToC()
134 se_base->payload = base.payload(); in ToC()
135 se_base->size = base.size(); in ToC()
139 SE_DeviceMemoryBase se_base; in ToC() local
140 ToC(base, &se_base); in ToC()
141 return se_base; in ToC()
144 stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base) { in FromC() argument
145 stream_executor::DeviceMemoryBase base(se_base.opaque, se_base.size); in FromC()
146 base.SetPayload(se_base.payload); in FromC()
Dc_api_conversions.h41 SE_DeviceMemoryBase* se_base);
42 stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base);
77 stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base);
/external/arm-trusted-firmware/plat/nvidia/tegra/include/drivers/
Dsecurity_engine.h34 const uint64_t se_base; member
/external/arm-trusted-firmware/plat/nvidia/tegra/soc/t210/drivers/se/
Dse_private.h638 return mmio_read_32(dev->se_base + offset); in tegra_se_read_32()
643 mmio_write_32(dev->se_base + offset, val); in tegra_se_write_32()
Dsecurity_engine.c93 .se_base = TEGRA_SE1_BASE,
108 .se_base = TEGRA_SE2_BASE,