Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 50 for constop (0.17 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

    }
    
    // Parses an R1 value to `shape` if it is a TF::ConstOp output. Otherwise,
    // returns an error.
    LogicalResult GetConstShapeValue(Value shape_value,
                                     llvm::SmallVector<int64_t, 8>* shape) {
      auto shape_op = shape_value.getDefiningOp();
      if (!shape_op) return failure();
      auto shape_const_op = llvm::dyn_cast<TF::ConstOp>(shape_op);
      if (!shape_const_op) return failure();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc

      return mlir::success();
    }
    
    // Creates a tf::ConcatOp that merges `input` values in `concat_dimension`.
    mlir::TF::ConcatOp CreateConcatOp(const int concat_dimension,
                                      const mlir::Location& location,
                                      mlir::ArrayRef<mlir::Value> inputs,
                                      mlir::OpBuilder* builder) {
      // Creates a const op to hold concat dimension value.
      auto concat_dim_type =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 21:28:13 UTC 2024
    - 34K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td

      "CreateInt32ConstOrCast($0, $_loc, $_builder)">;
    
    // Creates an int32 constant op from an integer attribute $0.
    def CreateInt32ConstOpFromIntAttr
      : NativeCodeCall<"$_builder.create<TF::ConstOp>($_loc, DenseElementsAttr::get(RankedTensorType::get({}, $_builder.getI32Type()), {static_cast<int32_t>($0.cast<IntegerAttr>().getInt())}))">;
    
    //===----------------------------------------------------------------------===//
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 28.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc

        auto func_inputs = reduce_func.getFunctionType().getInputs();
        const int func_input_size = func_inputs.size();
        const int argument_size =
            reduce_dataset->getAttrOfType<ArrayAttr>("Targuments").size();
        const int state_size =
            reduce_dataset->getAttrOfType<ArrayAttr>("Tstate").size();
        const int dataset_input_size = func_input_size - state_size - argument_size;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc

     public:
      FallbackToFlexOps() = default;
      explicit FallbackToFlexOps(const std::string &mode) { mode_ = mode; }
      FallbackToFlexOps(const FallbackToFlexOps &other) { mode_ = other.mode_; }
    
      void runOnOperation() override;
    
      StringRef getArgument() const final { return "quant-raise-flex-fallback"; }
    
      StringRef getDescription() const final {
        return "Fallback TF-Quantization-unsupported ops to TFLite Flex ops.";
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/hlo_matchers.cc

      }
    
      using StridedArrayViewBase::NextTensorIndex;
      using StridedArrayViewBase::size;
    
      int64_t operator[](int64_t i) const {
        return data_.getValues<APInt>()[OffsetForIndex(i)].getSExtValue();
      }
    
     private:
      const DenseIntElementsAttr& data_;
    };
    
    // It matches %iota generated from the following mlir codes:
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc

                                           Type result_type,
                                           Location conversion_loc) const final {
        if (!mlir::isa<TensorType>(result_type) ||
            !mlir::isa<TensorType>(input.getType()))
          return nullptr;
        return builder.create<TF::CastOp>(conversion_loc, result_type, input,
                                          /*truncate=*/builder.getBoolAttr(false));
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tfr/passes/decompose.cc

      explicit DecomposeTFOpsPass(std::optional<ModuleOp> external_tfr_module)
          : external_tfr_module_(external_tfr_module) {}
    
      StringRef getArgument() const final { return "tfr-decompose"; }
    
      StringRef getDescription() const final {
        return "Decompose TF ops with the registered composition library.";
      }
    
      void runOnOperation() override;
    
     private:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc

        if (!op->hasAttr(kMiniBatchSplitsAttr) && !op->hasAttr(kMiniBatchCsrAttr))
          return WalkResult::advance();
        Operation* defining = op->getOperand(0).getDefiningOp();
        if (llvm::dyn_cast_or_null<TF::ConstOp>(defining)) {
          op->emitError("Couldn't find a program key to insert into this op.");
          return WalkResult::interrupt();
        }
        return WalkResult::advance();
      });
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h

        return success();
      }
    
      LogicalResult processConstantOp(
          SourceOp op, Operation* const_op, int input_index,
          const operator_property::TensorProperty& tensor_property,
          PatternRewriter& rewriter) const {
        // Non-float tensors are neither weights nor require quantization.
        auto type = mlir::dyn_cast<ShapedType>(const_op->getResult(0).getType());
        if (!type || !mlir::isa<FloatType>(type.getElementType())) return success();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 28K bytes
    - Viewed (0)
Back to top