Searched defs:dims_b (Results 1 – 5 of 5) sorted by relevance
66 const int dims_b = rhs_shape.size(); in matchAndRewrite() local
242 const int dims_b = rhs_shape.size(); in matchAndRewrite() local
68 auto dims_b = shape_b.dimensions(); in CompareShapeDimsFromLeftToRight() local
170 const int dims_b = input_array_b.shape().dimensions_count(); in Run() local
122 const std::vector<int>& dims_b, in DescribeSparseTensorDenseMatMul()138 const std::vector<int>& dims_b, in DescribeXlaEinsum()155 const std::vector<int>& dims_b, in DescribeEinsum()593 const std::vector<int>& dims_b) { in DescribeBatchMatMul()