Searched defs:dims_b (Results 1 – 4 of 4) sorted by relevance
67 const int dims_b = rhs_shape.size(); in matchAndRewrite() local
68 auto dims_b = shape_b.dimensions(); in CompareShapeDimsFromLeftToRight() local
172 const int dims_b = input_array_b.shape().dimensions_count(); in Run() local
123 const std::vector<int>& dims_b, in DescribeSparseTensorDenseMatMul()139 const std::vector<int>& dims_b, in DescribeXlaEinsum()156 const std::vector<int>& dims_b, in DescribeEinsum()627 const std::vector<int>& dims_b) { in DescribeBatchMatMul()