Searched refs:b_input (Results 1 – 1 of 1) sorted by relevance
774 const auto& b_input = op_info.inputs(1); in CountBatchMatMulOperations() local794 b_input.shape(), std::max(matrix_rank, b_input.shape().dim_size()), in CountBatchMatMulOperations()799 (b_input.shape().dim_size() < matrix_rank); in CountBatchMatMulOperations()850 b_matrix->set_dtype(b_input.dtype()); in CountBatchMatMulOperations()