Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A 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()
H A Dop_kernel.h1239 AllocatorAttributes allocator_attr) { in allocate_tensor()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dscoped_allocator.h106 const AllocationAttributes& allocator_attr) override { in AllocateRaw()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
H A Dtrt_engine_op.cc543 AllocatorAttributes allocator_attr; in CopyToHostAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dkernels.cc754 tensorflow::AllocatorAttributes allocator_attr; in TF_AllocateTemp() local