| /external/tensorflow/tensorflow/core/kernels/ |
| l2loss_op_gpu.cu.cc | 52 typedef cub::TransformInputIterator<T, squareHalf<T>, T*> inputIterType;
57 functor::ReduceImpl<T, cub::Sum, T*, inputIterType, ReductionAxes>(
59 input.flat<T>().size(), 1, 1, 0, constants.kZero, cub::Sum());
|
| bincount_op_gpu.cu.cc | 20 #include "external/cub_archive/cub/device/device_histogram.cuh"
50 // In case weight.size() == 0, use CUB
62 auto err = cub::DeviceHistogram::HistogramEven(
85 err = cub::DeviceHistogram::HistogramEven(
|
| where_op_gpu.cu.h | 21 #include "external/cub_archive/cub/device/device_reduce.cuh"
22 #include "external/cub_archive/cub/device/device_select.cuh"
23 #include "external/cub_archive/cub/iterator/counting_input_iterator.cuh"
24 #include "external/cub_archive/cub/iterator/transform_input_iterator.cuh"
74 cub::TransformInputIterator<bool, IsNonzero<T>, const T*> is_nonzero_iter(
76 return cub::DeviceReduce::Sum(d_temp_storage, temp_storage_bytes,
88 return cub::DeviceReduce::Sum(d_temp_storage, temp_storage_bytes, d_in,
105 cub::CountingInputIterator<TIndex> select_counter(0);
107 cub::TransformInputIterator<bool, IsNonzero<T>, const T*> is_nonzero_iter(
109 return cub::DeviceSelect::Flagged
[all...] |
| softmax_op_gpu.cu.cc | 115 DoRowReduction<T, cub::Max, const T*>(
122 cub::CountingInputIterator<int> counting_iterator(0);
123 typedef cub::TransformInputIterator<T, SubtractAndExpFunctor<T>,
124 cub::CountingInputIterator<int>>
133 DoRowReduction<T, cub::Sum, InputIterType>(
|
| histogram_op_gpu.cu.cc | 21 #include "external/cub_archive/cub/device/device_histogram.cuh"
73 auto err = cub::DeviceHistogram::HistogramRange(
97 err = cub::DeviceHistogram::HistogramRange(
|
| reduction_gpu_kernels.cu.h | 21 #include "external/cub_archive/cub/device/device_reduce.cuh"
22 #include "external/cub_archive/cub/device/device_segmented_reduce.cuh"
23 #include "external/cub_archive/cub/iterator/counting_input_iterator.cuh"
24 #include "external/cub_archive/cub/iterator/transform_input_iterator.cuh"
25 #include "external/cub_archive/cub/warp/warp_reduce.cuh"
196 typedef cub::BlockReduce<value_type, num_threads> BlockReduce;
238 typedef cub::WarpReduce<value_type> WarpReduce;
284 cub::ShuffleIndex(sum, threadIdx.x + i * num_cols, 32, 0xffffffff);
370 typedef cub::WarpReduce<value_type> WarpReduce;
452 // handle situations where low latency is important better than CUB
[all...] |
| dynamic_partition_op_gpu.cu.cc | 19 // 2. We apply cub::DeviceRadixSort::SortPairs to the key - value pairs given
24 // - apply cub::DeviceReduce::ReduceByKey to count how many times each value
38 #include "external/cub_archive/cub/device/device_radix_sort.cuh"
39 #include "external/cub_archive/cub/device/device_reduce.cuh"
40 #include "external/cub_archive/cub/iterator/constant_input_iterator.cuh"
41 #include "external/cub_archive/cub/thread/thread_operators.cuh"
183 // R - the temporary storage used by cub::RadixSort, about 2N
340 cub::DeviceRadixSort::SortPairs(
351 cub::DeviceRadixSort::SortPairs(
398 cub::ConstantInputIterator<int32> values_in(1)
[all...] |
| topk_op_gpu.cu.cc | 23 #include "external/cub_archive/cub/device/device_segmented_radix_sort.cuh"
24 #include "external/cub_archive/cub/iterator/counting_input_iterator.cuh"
25 #include "external/cub_archive/cub/iterator/transform_input_iterator.cuh"
37 namespace cub { namespace
42 } // namespace cub
438 // TODO(ebrevdo): Once cub supports iterators for ValueT replace that tensor
447 cub::CountingInputIterator<int> counting_iter(0);
448 cub::TransformInputIterator<int, SegmentOffsetCreator,
449 cub::CountingInputIterator<int>>
471 auto err = cub::DeviceSegmentedRadixSort::SortPairsDescending
[all...] |
| depthwise_conv_op_gpu.cu.cc | 20 #include "external/cub_archive/cub/util_ptx.cuh"
[all...] |
| /external/mesa3d/src/gallium/auxiliary/gallivm/ |
| lp_bld_format_yuv.c | 233 LLVMValueRef cub; local
260 cub = lp_build_const_int_vec(gallivm, type, 516);
288 *b = LLVMBuildMul(builder, u, cub, "");
|
| /prebuilts/tools/common/m2/repository/com/puppycrawl/tools/checkstyle/5.9/ |
| checkstyle-5.9.jar | |
| /prebuilts/checkstyle/ |
| checkstyle.jar | |
| /external/webrtc/talk/media/testdata/ |
| h264-svc-99-640x360.rtpdump | [all...] |
