Searched refs:a_input (Results 1 – 1 of 1) sorted by relevance
773 const auto& a_input = op_info.inputs(0); in CountBatchMatMulOperations() local791 a_input.shape(), std::max(matrix_rank, a_input.shape().dim_size()), in CountBatchMatMulOperations()798 (a_input.shape().dim_size() < matrix_rank) || in CountBatchMatMulOperations()842 a_matrix->set_dtype(a_input.dtype()); in CountBatchMatMulOperations()