Searched defs:num_orows (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()196 const int64_t num_orows = c10::multiply_integers(sizes.begin(), sizes.begin() + dim); 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()397 const int64_t num_orows = c10::multiply_integers(sizes.begin(), sizes.begin() + dim); in scan_outer_dim() local