Searched refs:b_input_shape (Results 1 – 1 of 1) sorted by relevance
293 TensorShapeProto b_input_shape = MaybeGetMinimumShape( in IsEinsumCorrectlyFormed() local298 b_input_str.size() != static_cast<size_t>(b_input_shape.dim_size())) { in IsEinsumCorrectlyFormed()1158 TensorShapeProto b_input_shape = MaybeGetMinimumShape( in GenerateBatchMatmulContextFromEinsum() local1219 n_dim.set_size(n_dim.size() * b_input_shape.dim(i_idx).size()); in GenerateBatchMatmulContextFromEinsum()1276 TensorShapeProto b_input_shape = MaybeGetMinimumShape( in CountBatchMatMulOperations() local1288 TensorShapeProto* smaller_rank_shape = &b_input_shape; in CountBatchMatMulOperations()1289 if (b_input_shape.dim_size() > a_input_shape.dim_size()) { in CountBatchMatMulOperations()1290 bigger_rank_shape = &b_input_shape; in CountBatchMatMulOperations()1338 for (int i = std::max(0, b_input_shape.dim_size() - matrix_rank); in CountBatchMatMulOperations()1339 i < b_input_shape.dim_size(); ++i) { in CountBatchMatMulOperations()[all …]