Searched defs:alloc_attr (Results 1 – 15 of 15) sorted by relevance
30 AllocatorAttributes alloc_attr; in _HostConstantOp() local
252 AllocatorAttributes alloc_attr; in ComputeAsync() local288 AllocatorAttributes alloc_attr; in ComputeAsync() local
195 AllocatorAttributes alloc_attr; in Compute() local
196 AllocatorAttributes alloc_attr; in Compute() local916 AllocatorAttributes alloc_attr; in DoScatterNd() local
112 AllocatorAttributes alloc_attr; in Compute() local
128 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local145 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
134 AllocatorAttributes alloc_attr; member
83 AllocatorAttributes alloc_attr; in RecvOutputsFromRendezvousAsync() local
270 AllocatorAttributes* alloc_attr) { in ExtractScopedAllocatorAttr()
281 AllocatorAttributes* alloc_attr) { in ExtractScopedAllocatorAttr()
66 AllocatorAttributes alloc_attr; in ComputeSpansCore() local158 AllocatorAttributes alloc_attr; in ComputeGradSpansCore() local
277 AllocatorAttributes alloc_attr; in NmsGpu() local
828 AllocatorAttributes alloc_attr; in RunIfBoxIndexIsValid() local
100 AllocatorAttributes alloc_attr; in TransposeTensor() local
669 AllocatorAttributes alloc_attr; in GetDeviceAndAllocator() local