Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 25 for constant_ops (0.24 sec)

  1. tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc

                                               Type type, Location loc) {
      if (arith::ConstantOp::isBuildableWith(value, type))
        return builder.create<arith::ConstantOp>(loc, type,
                                                 value.cast<TypedAttr>());
      if (func::ConstantOp::isBuildableWith(value, type))
        return builder.create<func::ConstantOp>(loc, type,
                                                value.cast<FlatSymbolRefAttr>());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 21 16:55:41 UTC 2023
    - 38.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/python/representative_dataset_test.py

    from tensorflow.core.protobuf import meta_graph_pb2
    from tensorflow.python.client import session
    from tensorflow.python.framework import constant_op
    from tensorflow.python.framework import ops
    from tensorflow.python.framework import test_util
    from tensorflow.python.platform import test
    from tensorflow.python.types import core
    
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jan 04 07:35:19 UTC 2024
    - 11.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc

      auto constant_attr = rewriter->getI32TensorAttr(values);
      return rewriter->create<arith::ConstantOp>(loc, values_type, constant_attr);
    }
    
    // Creates ConstantOp for array of int64_t.
    arith::ConstantOp createI64ConstantOp(llvm::ArrayRef<int64_t> values,
                                          Location loc, PatternRewriter* rewriter) {
      auto values_type = RankedTensorType::get(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/utils/lstm_utils.cc

      return builder->create<arith::ConstantOp>(location, type, attr);
    }
    
    Value CreateF32SplatConst(OpBuilder* builder, ArrayRef<int64_t> shape,
                              float val, mlir::Location location) {
      auto type = RankedTensorType::get(shape, builder->getF32Type());
      auto attr = DenseElementsAttr::get(type, val);
      return builder->create<arith::ConstantOp>(location, type, attr);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 36.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc

      // It is legal to have TF ops in the graph still which can be
      // used later or in the case of SELECT where we allow TF ops in the final
      // graph.
      target.addLegalOp<mlir::arith::ConstantOp>();
      target.addLegalOp<mlir::func::ConstantOp>();
      target.addLegalOp<TFL::NoValueOp>();
      target.addLegalOp<ConstOp>();
      target.addLegalOp<DequantizeOp>();
      target.addLegalOp<QConstOp>();
      if (run_tfl_runtime_verification_) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 20:06:54 UTC 2024
    - 45.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfr/passes/decompose.cc

              attr_cst =
                  builder.create<ConstOp>(op->getLoc(), output_type, attribute);
            } else {
              attr_cst = builder.create<mlir::arith::ConstantOp>(
                  op->getLoc(), cast<TypedAttr>(attribute));
            }
            new_operands.push_back(attr_cst);
          }
        }
    
        // Create the TFR call op
        auto new_op = builder.create<CallOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.h

      // Duplicates the constant op if it has multiple uses, and replaces
      // target_op->operand[operand_index] with the newly created op. This also
      // replaces corresponsing quantization states.
      arith::ConstantOp DuplicateConstantOpIfNeeded(arith::ConstantOp op,
                                                    Operation* target_op,
                                                    int operand_index);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 20 11:42:17 UTC 2024
    - 16.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc

      // output gate bias is 0 since it is out of bounds of the bias tensor, so
      // we set its value as a const tensor of specified size and value 0.
      EXPECT_TRUE(
          mlir::cast<ElementsAttr>(mlir::cast<mlir::arith::ConstantOp>(
                                       it->getOpOperand(15).get().getDefiningOp())
                                       .getValue())
              .getValues<FloatAttr>()[0]
              .getValue()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h

          Value input = op.getOperand(index);
    
          if (input.getDefiningOp() == nullptr) continue;
    
          // TODO(b/172517537): make this work with non-PTQ case.
          if (llvm::isa<func::ConstantOp, arith::ConstantOp, TFL::ConstOp>(
                  input.getDefiningOp())) {
            // Tensors with derived scale are biases, and handled in propagation.
            if (tensor_property.use_derived_scale) continue;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 28K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc

      }
    
      auto scalar_type = RankedTensorType::get(/*shape=*/{}, fp_type);
      auto epsilon_tensor_attr = DenseElementsAttr::get(
          scalar_type, {mlir::cast<Attribute>(epsilon_attr)});
      Value epsilon = b.create<mhlo::ConstantOp>(epsilon_tensor_attr);
      auto dims_type = RankedTensorType::get(/*shape=*/{0}, b.getIntegerType(64));
      auto dims = DenseIntElementsAttr::get(dims_type, SmallVector<int64_t, 1>{});
      if (broadcast_to_type.hasStaticShape()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.2K bytes
    - Viewed (0)
Back to top