Searched refs:GetStepAllocator (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
H A Drenamed_device.h61 Allocator* GetStepAllocator(AllocatorAttributes attr,
63 return underlying_->GetStepAllocator(attr, step_resource_manager);
/external/tensorflow/tensorflow/core/framework/
H A Ddevice_base.h175 virtual Allocator* GetStepAllocator(AllocatorAttributes attr, function in class:tensorflow::DeviceBase
H A Dop_kernel.cc286 params_->device->GetStepAllocator(attr, resource_manager());
/external/tensorflow/tensorflow/contrib/nccl/kernels/
H A Dnccl_manager_test.cc65 return device->GetStepAllocator(AllocatorAttributes(),

Completed in 145 milliseconds