Home
last modified time | relevance | path

Searched defs:allocated_buffers_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.h54 std::vector<se::OwningDeviceMemory> allocated_buffers_; variable
Dgpu_conv_algorithm_picker.cc77 std::vector<se::OwningDeviceMemory> allocated_buffers_; member in xla::gpu::__anon559be1860111::ScratchAllocator
/external/tensorflow/tensorflow/stream_executor/gpu/
Dredzone_allocator.h124 std::vector<std::pair<OwningDeviceMemory, int64>> allocated_buffers_; variable
/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.h225 absl::flat_hash_set<const HloValue*> allocated_buffers_; variable