Searched defs:dims_a (Results 1 – 4 of 4) sorted by relevance
66 const int dims_a = lhs_shape.size(); in matchAndRewrite() local
67 auto dims_a = shape_a.dimensions(); in CompareShapeDimsFromLeftToRight() local
171 const int dims_a = input_array_a.shape().dimensions_count(); in Run() local
138 OpContext DescribeXlaEinsum(const std::vector<int>& dims_a, in DescribeXlaEinsum()155 OpContext DescribeEinsum(const std::vector<int>& dims_a, in DescribeEinsum()626 OpContext DescribeBatchMatMul(const std::vector<int>& dims_a, in DescribeBatchMatMul()