- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 141 for onStop (0.54 sec)
-
tensorflow/compiler/mlir/python/mlir_wrapper/ops.cc
.getOperation(); }); // mlir::TF::ConstOp py::class_<mlir::TF::ConstOp>(m, "Tf_ConstOp") .def("create", [](mlir::OpBuilder& opb, mlir::Location loc, mlir::Attribute value) -> mlir::Operation* { return opb.create<mlir::TF::ConstOp>(loc, value).getOperation(); }); // mlir::TF::EqualOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 30 02:12:49 UTC 2022 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/save_variables.cc
return ""; } auto assigned_value_operand = assign_var_op.getOperand(1); auto const_op = llvm::dyn_cast<mlir::TF::ConstOp>(assigned_value_operand.getDefiningOp()); if (!const_op) { assign_var_op->emitRemark( "Operand idx 1 is not a tf.ConstOp. The initializing tensor is not " "saved to checkpoint."); return ""; } Tensor const_tensor{};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:36:55 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/const_op_size.h
namespace mlir { namespace quant { // Returns the size in bytes of the underlying data of `const_op`. If the // underlying type's size cannot be determined, it assumes 4 bytes per element. int64_t GetSizeInBytes(TF::ConstOp const_op); } // namespace quant } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 14 04:37:13 UTC 2022 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
if (func.getNumArguments() == 0) { return; } auto builder = OpBuilder::atBlockBegin(&func.front()); llvm::SmallDenseMap<llvm::StringRef, TF::ConstOp> const_ops; llvm::BitVector arg_indexes_to_remove(func.getNumArguments()); // Replace arguments with const ops. for (BlockArgument argument : func.getArguments()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.td
include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td" include "mlir/IR/OpBase.td" include "mlir/Dialect/Arith/IR/ArithOps.td" def ConstDenseElementsI32ZeroAttr : NativeCodeCall<"$_builder.create<TFL::ConstOp>($_loc, DenseElementsAttr::get(RankedTensorType::get({}, $_builder.getI32Type()), {0}))">; def Size1InputRange : NativeCodeCall< "SmallVector<Value, 1>{$0}">; def Size2InputRange : NativeCodeCall<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 18 07:12:51 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.h
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" namespace mlir { namespace quant { std::optional<TF::PartitionedCallOp> ApplyUniformQuantization( PatternRewriter& rewriter, TF::ConstOp op, tensorflow::quantization::QuantizationComponentSpec& weight_spec); } // namespace quant } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Mar 24 07:44:40 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/prepare_quantize.cc
q_op->replaceAllUsesWith(new_qcast); return success(); } }; class ConvertTFConstOpToArithConstOp : public OpRewritePattern<TF::ConstOp> { public: using OpRewritePattern::OpRewritePattern; LogicalResult matchAndRewrite(TF::ConstOp op, PatternRewriter& rewriter) const override { rewriter.replaceOpWithNewOp<arith::ConstantOp>(op, op.getValue()); return success();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 05:11:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tfxla_device_specific_transforms.cc
tensorflow::Algorithm tensorflow_rng = tensorflow::ToTensorflowAlgorithm(xla_rng); OpBuilder opbuilder(get_alg_op); auto tf_const = opbuilder.create<TF::ConstOp>( get_alg_op->getLoc(), opbuilder.getI32IntegerAttr((int)tensorflow_rng)); get_alg_op->replaceAllUsesWith(tf_const); get_alg_op->erase(); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 05:56:39 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op_test.cc
quant_spec.set_tensor_type(QuantizationComponentSpec::TENSORTYPE_INT_8); std::optional<TF::PartitionedCallOp> dequantize_op = ApplyUniformQuantization( pattern_rewriter, cast<TF::ConstOp>(value.getDefiningOp()), quant_spec); EXPECT_TRUE(dequantize_op.has_value()); EXPECT_EQ(dequantize_op.value().func().getName().str(), "composite_dequantize_uniform"); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" namespace tflite { namespace { bool IsConst(mlir::Operation* op) { return llvm::isa<mlir::arith::ConstantOp, mlir::TF::ConstOp, mlir::TFL::ConstOp, mlir::TFL::QConstOp>(op); } bool IsOpSupported(mlir::Operation* op, const std::string& hardware) { auto* devce_hardware = mlir::TFL::tac::GetTargetHardware(hardware);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 06:11:34 UTC 2024 - 7.5K bytes - Viewed (0)