Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dhost_constant_op.cc30 AllocatorAttributes alloc_attr; in _HostConstantOp() local
Ddynamic_partition_op_gpu.cu.cc245 AllocatorAttributes alloc_attr; in ComputeAsync() local
281 AllocatorAttributes alloc_attr; in ComputeAsync() local
Dscale_and_translate_op.cc65 AllocatorAttributes alloc_attr; in ComputeSpansCore() local
157 AllocatorAttributes alloc_attr; in ComputeGradSpansCore() local
Dstack.cc194 AllocatorAttributes alloc_attr; in Compute() local
Dcrop_and_resize_op.cc786 AllocatorAttributes alloc_attr; in RunIfBoxIndexIsValid() local
Dtensor_array_ops.cc111 AllocatorAttributes alloc_attr; in Compute() local
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc102 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
113 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
Drendezvous_util.cc83 AllocatorAttributes alloc_attr; in RecvOutputsFromRendezvousAsync() local
Dexecutor.cc688 AllocatorAttributes* alloc_attr) { in ExtractScopedAllocatorAttr()
931 AllocatorAttributes alloc_attr; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc775 AllocatorAttributes alloc_attr; in GetDeviceAndAllocator() local