Home
last modified time | relevance | path

Searched defs:alloc_attr (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dpartitioning_utils.cc206 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
221 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
H A Dentry.h134 AllocatorAttributes alloc_attr; member
H A Drendezvous_util.cc83 AllocatorAttributes alloc_attr; in RecvOutputsFromRendezvousAsync() local
H A Dimmutable_executor_state.cc272 AllocatorAttributes* alloc_attr) { in ExtractScopedAllocatorAttr()
H A Dgraph_view.cc298 AllocatorAttributes* alloc_attr) { in ExtractScopedAllocatorAttr()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dscatter_nd_op.cc197 AllocatorAttributes alloc_attr; in Compute() local
871 AllocatorAttributes alloc_attr; in DoScatterNdOnCpu() local
965 AllocatorAttributes alloc_attr; in DoScatterNd() local
H A Ddynamic_partition_op_gpu.cu.cc241 AllocatorAttributes alloc_attr; in ComputeAsync() local
277 AllocatorAttributes alloc_attr; in ComputeAsync() local
H A Dhost_constant_op.cc30 AllocatorAttributes alloc_attr; in _HostConstantOp() local
H A Dstack.cc195 AllocatorAttributes alloc_attr; in Compute() local
H A Dtensor_array_ops.cc112 AllocatorAttributes alloc_attr; in Compute() local
H A Dresource_variable_ops.cc943 AllocatorAttributes alloc_attr; in CopyTensorToHost() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dscale_and_translate_op.cc66 AllocatorAttributes alloc_attr; in ComputeSpansCore() local
159 AllocatorAttributes alloc_attr; in ComputeGradSpansCore() local
H A Dcrop_and_resize_op.cc855 AllocatorAttributes alloc_attr; in RunIfBoxIndexIsValid() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dinfeed_ops.cc100 AllocatorAttributes alloc_attr; in TransposeTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_graph.cc680 AllocatorAttributes alloc_attr; in GetDeviceAndAllocator() local