Searched refs:b_inner_dim (Results 1 – 2 of 2) sorted by relevance
129 const int64 b_inner_dim = in ValidateInputs() local131 if (a_inner_dim != b_inner_dim) { in ValidateInputs()518 const int64 b_inner_dim = in Compute() local522 const int64 b_slice_size = b_inner_dim * b_outer_dim; in Compute()668 {b_inner_dim, b_outer_dim}); in Compute()
347 const int64 b_inner_dim = in Compute() local353 ctx, a_inner_dim == b_inner_dim, in Compute()