Searched defs:result_strides (Results 1 – 5 of 5) sorted by relevance
514 IntArrayRef result_strides = result_->strides(); in block_sparse_mm() local
525 IntArrayRef result_strides = result.strides(); in baddbmm_out_cuda_impl() local
472 …auto result_strides = at::native::batched_matrix_contiguous_strides(B_broadcast_size, /*f_contig=*… in TORCH_META_FUNC() local528 …auto result_strides = at::native::batched_matrix_contiguous_strides(result_shape, /*f_contig=*/lef… in TORCH_META_FUNC() local552 auto result_strides = at::native::batched_matrix_contiguous_strides(shape, /*f-contig*=*/true); in TORCH_META_FUNC() local607 …auto result_strides = at::native::batched_matrix_contiguous_strides(B_broadcast_size, /*f_contig=*… in TORCH_META_FUNC() local
1415 const auto result_strides = result.strides(); in addmm_impl_cpu_() local1688 const auto result_strides = result.strides(); in baddbmm_with_gemm_() local
914 auto result_strides = result_nonzero.strides().vec(); in TORCH_IMPL_FUNC() local