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()
|