Home
last modified time | relevance | path

Searched refs:SE_MaybeOwningDeviceMemory (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h173 typedef struct SE_MaybeOwningDeviceMemory { struct
180 } SE_MaybeOwningDeviceMemory; argument
247 SE_MaybeOwningDeviceMemory* buffers;
264 SE_MaybeOwningDeviceMemory* to_be_released;
Dc_api_conversions.h91 SE_MaybeOwningDeviceMemory* se_mem,
98 SE_MaybeOwningDeviceMemory ToC(stream_executor::OwningDeviceMemory* mem);
102 SE_MaybeOwningDeviceMemory ToC(xla::MaybeOwningDeviceMemory& mem, bool aliased);
Dc_api_conversions.cc43 SE_MaybeOwningDeviceMemory ToC(xla::MaybeOwningDeviceMemory& mem, in ToC()
45 SE_MaybeOwningDeviceMemory se_mem; in ToC()
66 SE_MaybeOwningDeviceMemory* se_mem, in FromC()
122 SE_MaybeOwningDeviceMemory ToC(stream_executor::OwningDeviceMemory* mem) { in ToC()
123 SE_MaybeOwningDeviceMemory se_mem; in ToC()
Dtpu_executor_c_api.h323 SE_MaybeOwningDeviceMemory* array);
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc111 absl::InlinedVector<SE_MaybeOwningDeviceMemory, 2> se_buffers; in ExecuteAsyncOnStream()
117 new SE_MaybeOwningDeviceMemory[se_buffers.size()]; in ExecuteAsyncOnStream()