Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dhost_constant_op.cc30 AllocatorAttributes alloc_attr; in _HostConstantOp() local
Ddynamic_partition_op_gpu.cu.cc241 AllocatorAttributes alloc_attr; in ComputeAsync() local
277 AllocatorAttributes alloc_attr; in ComputeAsync() local
Dstack.cc195 AllocatorAttributes alloc_attr; in Compute() local
Dscatter_nd_op.cc196 AllocatorAttributes alloc_attr; in Compute() local
916 AllocatorAttributes alloc_attr; in DoScatterNd() local
Dtensor_array_ops.cc112 AllocatorAttributes alloc_attr; in Compute() local
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc125 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
142 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
Dentry.h134 AllocatorAttributes alloc_attr; member
Drendezvous_util.cc83 AllocatorAttributes alloc_attr; in RecvOutputsFromRendezvousAsync() local
Dimmutable_executor_state.cc271 AllocatorAttributes* alloc_attr) { in ExtractScopedAllocatorAttr()
Dgraph_view.cc281 AllocatorAttributes* alloc_attr) { in ExtractScopedAllocatorAttr()
/external/tensorflow/tensorflow/core/kernels/image/
Dscale_and_translate_op.cc66 AllocatorAttributes alloc_attr; in ComputeSpansCore() local
159 AllocatorAttributes alloc_attr; in ComputeGradSpansCore() local
Dcrop_and_resize_op.cc853 AllocatorAttributes alloc_attr; in RunIfBoxIndexIsValid() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dinfeed_ops.cc100 AllocatorAttributes alloc_attr; in TransposeTensor() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc684 AllocatorAttributes alloc_attr; in GetDeviceAndAllocator() local