Home
last modified time | relevance | path

Searched defs:cpu_alloc (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.cc142 AllocatorAttributes cpu_alloc; in WriteOptionalWithValueToOutput() local
153 AllocatorAttributes cpu_alloc; in WriteOptionalNoneToOutput() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dzeros_op.cc54 AllocatorAttributes cpu_alloc; in Compute() local
Dsparse_tensor_to_csr_sparse_matrix_op.cc103 AllocatorAttributes cpu_alloc; in Compute() local
270 AllocatorAttributes cpu_alloc; in ComputeAsync() local
Ddense_to_csr_sparse_matrix_op.cc124 AllocatorAttributes cpu_alloc; in Compute() local
337 AllocatorAttributes cpu_alloc; in ComputeAsync() local
Dsparse_cholesky_op.cc219 AllocatorAttributes cpu_alloc; in Compute() local
Dsparse_mat_mul_op.cc253 AllocatorAttributes cpu_alloc; in Compute() local
/external/tensorflow/tensorflow/core/framework/
Dallocator.cc69 static Allocator* cpu_alloc = in cpu_allocator_base() local
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.cc197 AllocatorAttributes cpu_alloc; in RegisterInput() local