Home
last modified time | relevance | path

Searched defs:allocator_attr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dscoped_allocator.h102 const AllocationAttributes& allocator_attr) override { in AllocateRaw()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h971 Tensor* out_temp, AllocatorAttributes allocator_attr) { in allocate_temp()
1241 AllocatorAttributes allocator_attr) { in allocate_tensor()
Dop_kernel.cc566 const TensorShape& shape, const AllocatorAttributes& allocator_attr, in forward_input_or_allocate_temp()
744 AllocatorAttributes allocator_attr, in allocate_temp()