Searched refs:device_memory_ (Results 1 – 5 of 5) 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
174 return device_memory_; in device_memory()177 return device_memory_; in device_memory()189 void ReleaseDeviceMemory() { device_memory_.clear(); } in ReleaseDeviceMemory()226 absl::InlinedVector<se::DeviceMemoryBase, 1> device_memory_; variable
127 for (const se::DeviceMemoryBase& buf : device_memory_) { in AsShapedBuffer()142 for (const se::DeviceMemoryBase& buf : device_memory_) { in AddToInputAsImmutable()155 for (const se::DeviceMemoryBase& buf : device_memory_) { in AddToInputAsDonated()172 device_memory_(device_memory.begin(), device_memory.end()), in TrackedDeviceBuffer()180 for (const se::DeviceMemoryBase& buffer : device_memory_) { in ~TrackedDeviceBuffer()
76 device_memory_(nullptr) {} in ContextDeviceMemory()79 if (device_memory_) { in ~ContextDeviceMemory()80 device_memory_allocator_->free(device_memory_); in ~ContextDeviceMemory()89 device_memory_ = nullptr; in AllocateDeviceMemory()92 device_memory_ = device_memory_allocator_->allocate( in AllocateDeviceMemory()95 if (device_memory_ == nullptr) { in AllocateDeviceMemory()103 execution_context_->setDeviceMemory(device_memory_); in AllocateDeviceMemory()111 void* device_memory_; member in tensorflow::tensorrt::__anon7713a9350111::ContextDeviceMemory