Searched refs:b_input (Results 1 – 1 of 1) sorted by relevance
272 const auto& b_input = op_info.inputs(1); in IsEinsumCorrectlyFormed() local294 b_input.shape(), std::max(kMatrixRank, b_input.shape().dim_size()), in IsEinsumCorrectlyFormed()1145 const auto& b_input = op_info.inputs(1); in GenerateBatchMatmulContextFromEinsum() local1159 b_input.shape(), std::max(kMatrixRank, b_input.shape().dim_size()), in GenerateBatchMatmulContextFromEinsum()1164 (b_input.shape().dim_size() < kMatrixRank); in GenerateBatchMatmulContextFromEinsum()1180 b_matrix->set_dtype(b_input.dtype()); in GenerateBatchMatmulContextFromEinsum()1257 const auto& b_input = op_info.inputs(1); in CountBatchMatMulOperations() local1277 b_input.shape(), std::max(matrix_rank, b_input.shape().dim_size()), in CountBatchMatMulOperations()1282 (b_input.shape().dim_size() < matrix_rank); in CountBatchMatMulOperations()1336 b_matrix->set_dtype(b_input.dtype()); in CountBatchMatMulOperations()