Searched defs:outer_offset (Results 1 – 3 of 3) sorted by relevance
97 const int64_t outer_offset, in segment_reduce_forward_kernel()168 const int64_t outer_offset, in segment_reduce_backward_kernel()291 int64_t outer_offset = 1, inner_offset = 1; in _segment_reduce_lengths_offsets_backward_cuda_kernel() local419 int64_t outer_offset = 1, inner_offset = 1; in _segment_reduce_lengths_offsets_cuda_kernel() local
265 const index_t outer_offset = outer_index * outer_stride; in cunn_SpatialSoftMaxForward() local323 const uint32_t outer_offset = outer_index * outer_stride; in cunn_SpatialSoftMaxBackward() local
42 int64_t outer_offset = 1, inner_offset = 1; in _segment_reduce_lengths_cpu_kernel1() local196 int64_t outer_offset = 1, inner_offset = 1; in _segment_reduce_cpu_lengths_backward_kernel1() local