Searched refs:device_memory_ (Results 1 – 4 of 4) sorted by relevance
23 parent_->temporary_memory_manager()->MarkFinalized(device_memory_, in ~TemporaryDeviceMemoryBase()31 return &device_memory_; in mutable_device_memory()37 return device_memory_; in device_memory()42 parent_->temporary_memory_manager()->MarkFinalized(device_memory_, in Finalize()49 device_memory_, allocation_generation_); in IsFinalized()54 device_memory_, allocation_generation_); in IsAllocated()60 : device_memory_(device_memory), in TemporaryDeviceMemoryBase()
97 DeviceMemoryBase device_memory_; variable
49 : device_memory_(executor, executor->AllocateArray<uint8>(length)), in InfeedBuffer()51 CHECK(!device_memory_->is_null()); in InfeedBuffer()56 se::DeviceMemoryBase* device_memory() { return device_memory_.ptr(); } in device_memory()59 se::ScopedDeviceMemory<uint8> device_memory_;
48 device_memory_(buffer_, length_) {} in CpuInfeedBuffer()55 se::DeviceMemoryBase* device_memory() { return &device_memory_; } in device_memory()60 se::DeviceMemoryBase device_memory_; member in xla::__anonf0769de40111::CpuInfeedBuffer