Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for getRhsContractingDimensions (0.25 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

                     << filter_type << "\n");
          return failure();
        }
        const int rhs_rank = filter_type.cast<TensorType>().getRank();
        const auto rhs_contracting_dim =
            dot_dimension_nums.getRhsContractingDimensions()[0];
        if ((rhs_contracting_dim != rhs_rank - 1) &&
            (rhs_contracting_dim != rhs_rank - 2)) {
          LLVM_DEBUG(llvm::dbgs()
                     << "Not supported rhs contracting dim for dot_general.\n");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

          dot_dimension_numbers.getLhsContractingDimensions());
      DotDimensionsInfo rhs_dot_dimensions_info(
          rhs_type, dot_dimension_numbers.getRhsBatchingDimensions(),
          dot_dimension_numbers.getRhsContractingDimensions());
    
      // Transposes lhs shape to be in the order of {batch_dimensions,
      // out_dimensions, contracting dimensions}.
      llvm::SmallVector<int64_t, 4> lhs_permutation = Concat<int64_t>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

            auto rhs_contracting_dimensions = builder_.CreateVector(
                mlir::GetVector<int64_t>(mlir::cast<mlir::vhlo::TensorV1Attr>(
                                             vhlo_op.getRhsContractingDimensions()),
                                         vhlo_type_converter));
    
            BufferOffset<flatbuffers::Vector<unsigned int>> precision_config = 0;
            if (vhlo_op.getPrecisionConfig()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
Back to top