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.cc241 AllocatorAttributes alloc_attr; in ComputeAsync() local
277 AllocatorAttributes alloc_attr; in ComputeAsync() local
Dscatter_nd_op.cc197 AllocatorAttributes alloc_attr; in Compute() local
871 AllocatorAttributes alloc_attr; in DoScatterNdOnCpu() local
965 AllocatorAttributes alloc_attr; in DoScatterNd() local
Dstack.cc195 AllocatorAttributes alloc_attr; in Compute() local
Dtensor_array_ops.cc112 AllocatorAttributes alloc_attr; in Compute() local
Dresource_variable_ops.cc943 AllocatorAttributes alloc_attr; in CopyTensorToHost() local
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc206 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
221 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
Dentry.h134 AllocatorAttributes alloc_attr; member
Drendezvous_util.cc83 AllocatorAttributes alloc_attr; in RecvOutputsFromRendezvousAsync() local
Dimmutable_executor_state.cc272 AllocatorAttributes* alloc_attr) { in ExtractScopedAllocatorAttr()
Dgraph_view.cc298 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.cc855 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.cc680 AllocatorAttributes alloc_attr; in GetDeviceAndAllocator() local