Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h110 typedef struct SE_DeviceMemoryAllocator { struct
115 } SE_DeviceMemoryAllocator; argument
163 SE_DeviceMemoryAllocator allocator;
179 SE_DeviceMemoryAllocator allocator;
Dtpu_executor_c_api.h290 SE_StreamExecutor* stream_executor, SE_DeviceMemoryAllocator* allocator,
295 SE_StreamExecutor* stream_executor, SE_DeviceMemoryAllocator* allocator,
301 SE_DeviceMemoryAllocator* allocator, SE_Executable** executables,
Dc_api_conversions.h95 SE_DeviceMemoryAllocator ToC(stream_executor::DeviceMemoryAllocator* allocator);
Dc_api_conversions.cc77 SE_DeviceMemoryAllocator ToC( in ToC()
79 SE_DeviceMemoryAllocator se_allocator; in ToC()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc309 SE_DeviceMemoryAllocator allocator = in Compile()