Searched refs:b_rank (Results 1 – 2 of 2) sorted by relevance
56 const int b_rank = b_t.dims(); in Compute() local61 if (b_rank == 3) { in Compute()73 OP_REQUIRES(ctx, b_rank == 0, in Compute()82 if (b_rank == 0) { in Compute()
2758 auto b_rank = b_type.getRank(); in Verify() local2759 if (a_rank != b_rank) in Verify()2765 b_type.getDimSize(b_rank - (op.left_side() ? 2 : 1))) in Verify()