Searched defs:can_use_32bit_indexing (Results 1 – 4 of 4) sorted by relevance
1157 bool can_use_32bit_indexing = iter.can_use_32bit_indexing(); in gpu_reduce_kernel() local1264 bool can_use_32bit_indexing = iter.can_use_32bit_indexing(); in jitted_gpu_reduce_kernel() local
790 …const bool can_use_32bit_indexing = is_32bit_representable(shift) && is_32bit_representable(output… in cunn_SoftMaxBackward() local
1169 def can_use_32bit_indexing( member in SIMDScheduling
1293 bool TensorIteratorBase::can_use_32bit_indexing() const { in can_use_32bit_indexing() function in at::TensorIteratorBase