Searched refs:cpu_alloc (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | allocator.cc | 69 static Allocator* cpu_alloc = in cpu_allocator_base() local 75 if (cpu_allocator_collect_full_stats && !cpu_alloc->TracksAllocationSizes()) { in cpu_allocator_base() 76 cpu_alloc = new TrackingAllocator(cpu_alloc, true); in cpu_allocator_base() 78 return cpu_alloc; in cpu_allocator_base()
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | optional_ops.cc | 142 AllocatorAttributes cpu_alloc; in WriteOptionalWithValueToOutput() local 143 cpu_alloc.set_on_host(true); in WriteOptionalWithValueToOutput() 145 &variant_t, cpu_alloc)); in WriteOptionalWithValueToOutput() 153 AllocatorAttributes cpu_alloc; in WriteOptionalNoneToOutput() local 154 cpu_alloc.set_on_host(true); in WriteOptionalNoneToOutput() 156 &variant_t, cpu_alloc)); in WriteOptionalNoneToOutput()
|
/external/tensorflow/tensorflow/core/kernels/sparse/ |
D | zeros_op.cc | 54 AllocatorAttributes cpu_alloc; in Compute() local 55 cpu_alloc.set_on_host(true); in Compute() 57 c, c->allocate_output(0, TensorShape({}), &matrix_t, cpu_alloc)); in Compute()
|
D | sparse_tensor_to_csr_sparse_matrix_op.cc | 103 AllocatorAttributes cpu_alloc; in Compute() local 104 cpu_alloc.set_on_host(true); in Compute() 107 cpu_alloc)); in Compute() 270 AllocatorAttributes cpu_alloc; in ComputeAsync() local 271 cpu_alloc.set_on_host(true); in ComputeAsync() 273 c, c->allocate_output(0, TensorShape({}), &matrix_t, cpu_alloc), in ComputeAsync()
|
D | dense_to_csr_sparse_matrix_op.cc | 124 AllocatorAttributes cpu_alloc; in Compute() local 125 cpu_alloc.set_on_host(true); in Compute() 128 cpu_alloc)); in Compute() 337 AllocatorAttributes cpu_alloc; in ComputeAsync() local 338 cpu_alloc.set_on_host(true); in ComputeAsync() 340 c, c->allocate_output(0, TensorShape({}), &matrix_t, cpu_alloc), in ComputeAsync()
|
D | sparse_cholesky_op.cc | 219 AllocatorAttributes cpu_alloc; in Compute() local 220 cpu_alloc.set_on_host(true); in Compute() 223 cpu_alloc)); in Compute()
|
D | sparse_mat_mul_op.cc | 253 AllocatorAttributes cpu_alloc; in Compute() local 254 cpu_alloc.set_on_host(true); in Compute() 257 cpu_alloc)); in Compute()
|
/external/tensorflow/tensorflow/core/kernels/batching_util/ |
D | batch_resource_base.cc | 197 AllocatorAttributes cpu_alloc; in RegisterInput() local 198 cpu_alloc.set_on_host(true); in RegisterInput() 200 &empty_output, cpu_alloc)); in RegisterInput()
|