Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Doptional_ops.cc142 AllocatorAttributes cpu_alloc; in WriteOptionalWithValueToOutput() local
153 AllocatorAttributes cpu_alloc; in WriteOptionalNoneToOutput() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dsparse_tensor_to_csr_sparse_matrix_op.cc119 AllocatorAttributes cpu_alloc; in Compute() local
286 AllocatorAttributes cpu_alloc; in ComputeAsync() local
H A Ddense_to_csr_sparse_matrix_op.cc124 AllocatorAttributes cpu_alloc; in Compute() local
338 AllocatorAttributes cpu_alloc; in ComputeAsync() local
H A Dzeros_op.cc54 AllocatorAttributes cpu_alloc; in Compute() local
H A Dsparse_cholesky_op.cc221 AllocatorAttributes cpu_alloc; in Compute() local
H A Dsparse_mat_mul_op.cc253 AllocatorAttributes cpu_alloc; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dallocator.cc67 static Allocator* cpu_alloc = in cpu_allocator_base() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbatch_resource_base.cc294 AllocatorAttributes cpu_alloc; in RegisterInput() local