Searched refs:hlo_dims (Results 1 – 2 of 2) sorted by relevance
667 auto hlo_dims = config.dot_dimension_numbers(); in EmitGemm() local669 GetI64DenseElementsAttr(hlo_dims.lhs_batch_dimensions()), in EmitGemm()670 GetI64DenseElementsAttr(hlo_dims.rhs_batch_dimensions()), in EmitGemm()671 GetI64DenseElementsAttr(hlo_dims.lhs_contracting_dimensions()), in EmitGemm()672 GetI64DenseElementsAttr(hlo_dims.rhs_contracting_dimensions()), in EmitGemm()
1264 auto hlo_dims = hlo_shape.dimensions(); in SliceFirstK() local1266 std::vector<int64> limit_indices(hlo_dims.begin(), hlo_dims.end()); in SliceFirstK()