Searched defs:batch_dimensions (Results 1 – 4 of 4) sorted by relevance
395 DotDimensionsInfo(ShapedType type, DenseIntElementsAttr batch_dimensions, in DotDimensionsInfo()418 const DimensionVector &batch_dimensions() const { return batch_dimensions_; } in batch_dimensions() function in mlir::TF::__anon0527096c0111::DotDimensionsInfo
494 std::vector<int64> batch_dimensions; in BuildTriangularSolve() local
397 HloInstruction* dot_operand, absl::Span<const int64> batch_dimensions, in NormalizeDotOperandToBatchMajorAndContractingMinor()
542 auto batch_dimensions = GetI64ElementsAttr( in BatchDot() local2851 auto batch_dimensions = GetI64ElementsAttr( in matchAndRewrite() local