HomeSort by relevance Sort by last modified time
    Searched defs:cub (Results 1 - 2 of 2) sorted by null

  /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, "");
  /external/tensorflow/tensorflow/core/kernels/
topk_op_gpu.h 25 #include "third_party/cub/device/device_segmented_radix_sort.cuh"
26 #include "third_party/cub/iterator/counting_input_iterator.cuh"
27 #include "third_party/cub/iterator/transform_input_iterator.cuh"
39 namespace cub { namespace
44 } // namespace cub
441 // TODO(ebrevdo): Once cub supports iterators for ValueT replace that tensor
450 cub::CountingInputIterator<int> counting_iter(0);
451 cub::TransformInputIterator<int, SegmentOffsetCreator,
452 cub::CountingInputIterator<int>>
474 auto err = cub::DeviceSegmentedRadixSort::SortPairsDescending
    [all...]

Completed in 376 milliseconds