Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 22 for constant_ops (0.2 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc

        auto new_shape_const_attr =
            DenseElementsAttr::get(shape_spec_type, new_shape.getShape());
        rewriter.setInsertionPointAfter(weight_op);
        auto new_shape_const = rewriter.create<arith::ConstantOp>(
            weight_op->getLoc(), shape_spec_type, new_shape_const_attr);
        auto reshape_op = rewriter.create<TF::ReshapeOp>(
            weight_op->getLoc(), new_shape, weight_op->getResult(0),
            new_shape_const);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc

          scales, zero_points, quant_dim, qtype.getStorageTypeMin(),
          qtype.getStorageTypeMax());
    }
    
    }  // namespace
    
    bool IsOpQuantizable(Operation* op) {
      if (isa<func::ConstantOp, arith::ConstantOp, quantfork::StatisticsOp>(op)) {
        // Constant ops do not have QuantizableResult attribute but they can deal
        // with quantized tensors.
        return true;
      } else if (op->hasTrait<OpTrait::IsTerminator>() ||
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 43.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

      }
    
      auto dense_attr_or = GetDenseAttrFromTensorProtoAttr(
          tensor_proto_attr.getValue(), new_operand_type);
      if (failed(dense_attr_or)) return failure();
    
      return Value(rewriter.create<mhlo::ConstantOp>(op->getLoc(), new_operand_type,
                                                     *dense_attr_or));
    }
    
    xla::ConvolutionDimensionNumbers ConvertConvolutionDimensionNumbers(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
Back to top