Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dhost_constant_op.cc30 AllocatorAttributes alloc_attr; in _HostConstantOp() local
Ddynamic_partition_op_gpu.cu.cc252 AllocatorAttributes alloc_attr; in ComputeAsync() local
288 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.cc128 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
145 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
Dentry.h134 AllocatorAttributes alloc_attr; member
Drendezvous_util.cc83 AllocatorAttributes alloc_attr; in RecvOutputsFromRendezvousAsync() local
Dimmutable_executor_state.cc270 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
158 AllocatorAttributes alloc_attr; in ComputeGradSpansCore() local
Dnon_max_suppression_op.cu.cc277 AllocatorAttributes alloc_attr; in NmsGpu() local
Dcrop_and_resize_op.cc828 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.cc669 AllocatorAttributes alloc_attr; in GetDeviceAndAllocator() local