Searched refs:temp_storage (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_gpu_kernels.cu.h | 186 __shared__ typename BlockReduce::TempStorage temp_storage;
195 sum = BlockReduce(temp_storage).Reduce(sum, op, num_elements_to_reduce);
232 __shared__ typename WarpReduce::TempStorage temp_storage;
235 WarpReduce(temp_storage).Reduce(sum, op, min(num_cols, TF_RED_WARPSIZE));
417 __shared__ typename WarpReduce::TempStorage temp_storage;
421 WarpReduce(temp_storage).HeadSegmentedReduce(val, head_flag, op);
675 Tensor temp_storage;
680 &temp_storage));
684 (T*)temp_storage.flat<int8_t>().data(), in_size,
693 (T*)temp_storage.flat<int8_t>().data(), out, 1,
[all …]
|
| D | bincount_op_gpu.cu.cc | 87 Tensor temp_storage; in Compute() local
90 TensorShape({static_cast<int64>(temp_storage_bytes)}), &temp_storage)); in Compute()
92 void* d_temp_storage = temp_storage.flat<int8>().data(); in Compute()
|
| D | histogram_op_gpu.cu.cc | 98 Tensor temp_storage; in Compute() local
101 TensorShape({static_cast<int64>(temp_storage_bytes)}), &temp_storage)); in Compute()
103 void* d_temp_storage = temp_storage.flat<int8>().data(); in Compute()
|
| D | where_op_gpu.cu.h | 175 Tensor temp_storage;
178 &temp_storage));
181 /*temp_storage*/ temp_storage.flat<int8>().data(), temp_storage_bytes,
312 Tensor temp_storage;
315 &temp_storage));
318 /*temp_storage*/ temp_storage.flat<int8>().data(), temp_storage_bytes,
|
| D | scan_ops_gpu.h | 201 } temp_storage; in __launch_bounds__() local
234 BlockLoad(temp_storage.load).Load(permutein_iter, thread_data, valid_items); in __launch_bounds__()
239 BlockScan(temp_storage.scan) in __launch_bounds__()
242 BlockScan(temp_storage.scan) in __launch_bounds__()
248 BlockStore(temp_storage.store) in __launch_bounds__()
|
| D | topk_op_gpu.h | 497 Tensor temp_storage;
500 &temp_storage));
502 /* d_temp_storage */ temp_storage.flat<int8>().data(),
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 108 Tensor temp_storage; in operator ()() local
111 &temp_storage)); in operator ()()
112 DCHECK_NE(temp_storage.flat<int8>().data(), nullptr); in operator ()()
114 /*d_temp_storage*/ temp_storage.flat<int8>().data(), in operator ()()
|
| /external/libvpx/libvpx/vp8/common/x86/ |
| D | loopfilter_block_sse2_x86_64.asm | 355 %define temp_storage 0 ; size is 256 (16*16)
|