Searched refs:b_dim (Results 1 – 2 of 2) sorted by relevance
297 b_dim = np_utils.getitem(array_ops.shape(b), -1)314 b = maybe_pad_0(b, b_dim)329 (a_dim == 2) & (b_dim == 2),
1297 int b_dim = bigger_rank_shape->dim(b_i).size(); in CountBatchMatMulOperations() local1305 num_matmuls *= std::max(b_dim, s_dim); in CountBatchMatMulOperations()