Searched defs:self_stride (Results 1 – 7 of 7) sorted by relevance
16 int64_t self_stride = self.stride(dim); in tensor_dim_apply3() local
90 auto self_stride = (self.dim() > 2 && self.stride(-3) > 0) ? self.stride(-3) : 1; in apply_triu_tril() local
1421 auto self_stride = self.dim() == 0 ? 1 : self.stride(dim); in index_select_out_cpu_() local1440 auto self_stride = self.dim() == 0 ? 1 : self.stride(dim); in index_select_out_cpu_() local
773 int self_dim_size, int self_stride, int values_stride, int indices_stride) { in cummax_cummin_helper()
96 int64_t self_stride = in tril_kernel() local
365 auto self_stride = get_stride_for_index(self, i); in bmm_nested_cuda() local
1469 const auto self_stride = matrixStride(self); in lu_factor_looped_cusolver() local