Searched defs:batch_dimensions (Results 1 – 5 of 5) sorted by relevance
220 auto validate_matrix = [&](const Shape &shape, auto batch_dimensions) { in RunGemm()
494 std::vector<int64> batch_dimensions; in BuildTriangularSolve() local
433 HloInstruction* dot_operand, absl::Span<const int64> batch_dimensions, in NormalizeDotOperandToBatchMajorAndContractingMinor()
692 DotDimensionsInfo(ShapedType type, DenseIntElementsAttr batch_dimensions, in DotDimensionsInfo()715 const DimensionVector &batch_dimensions() const { return batch_dimensions_; } in batch_dimensions() function in mlir::TF::__anon964666f10111::DotDimensionsInfo
492 auto batch_dimensions = GetI64ElementsAttr( in BatchDot() local3459 auto batch_dimensions = GetI64ElementsAttr( in matchAndRewrite() local