Searched defs:num_irows (Results 1 – 1 of 1) sorted by relevance
156 …const uint32_t num_orows, const uint32_t num_irows, const uint32_t row_size, scalar_t init, Binary… in tensor_kernel_scan_outer_dim_with_indices()199 const int64_t num_irows = c10::multiply_integers(sizes.begin() + dim + 1, sizes.end()); in scan_outer_dim_with_indices() local273 … const uint32_t num_orows, const uint32_t num_irows, const uint32_t row_size, in tensor_kernel_scan_outer_dim()400 const int64_t num_irows = c10::multiply_integers(sizes.begin() + dim + 1, sizes.end()); in scan_outer_dim() local