Home
last modified time | relevance | path

Searched defs:allocation_attr (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtyped_allocator.h38 const AllocationAttributes& allocation_attr) { in Allocate()
Dallocator.h202 const AllocationAttributes& allocation_attr) { in AllocateRaw()
311 const AllocationAttributes& allocation_attr) override { in AllocateRaw()
Dtracking_allocator.cc34 const AllocationAttributes& allocation_attr) { in AllocateRaw()
Dtensor.cc477 const AllocationAttributes& allocation_attr) in Buffer()
778 const AllocationAttributes& allocation_attr) in Tensor()
Dop_kernel.cc708 AllocatorAttributes attr, const AllocationAttributes& allocation_attr) { in allocate_tensor()
778 const AllocationAttributes& allocation_attr) { in allocate_temp()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h70 AllocationAttributes allocation_attr; in AllocateBytes() local
Dfft_ops.cc374 AllocationAttributes allocation_attr; in AllocateBytes() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding.cc72 const AllocationAttributes& allocation_attr) { in InitPartial()
Dbase_rendezvous_mgr.cc268 AllocationAttributes allocation_attr; in SameWorkerRecvDone() local
/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.cc228 const AllocationAttributes& allocation_attr) { in AllocateRawInternalWithRetry()
254 const AllocationAttributes& allocation_attr) { in AllocateRaw()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc648 AllocationAttributes allocation_attr; in MaybeCopyTensorToGPU() local