Home
last modified time | relevance | path

Searched defs:allocator_attr (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dscoped_allocator.h106 const AllocationAttributes& allocator_attr) override { in AllocateRaw()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc311 AllocatorAttributes allocator_attr) { in allocate_temp()
581 const TensorShape& shape, const AllocatorAttributes& allocator_attr, in forward_input_or_allocate_temp()
822 AllocatorAttributes allocator_attr, in allocate_temp()
860 AllocatorAttributes allocator_attr) { in allocate_temp()
Dop_kernel.h1239 AllocatorAttributes allocator_attr) { in allocate_tensor()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc543 AllocatorAttributes allocator_attr; in CopyToHostAsync() local
/external/tensorflow/tensorflow/c/
Dkernels.cc754 tensorflow::AllocatorAttributes allocator_attr; in TF_AllocateTemp() local