- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 201 for constop (0.4 sec)
-
tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_mlrt_ops.td
}]; let arguments = (ins StrAttr:$node_def, I32Attr:$op_key ); let assemblyFormat = "attr-dict"; } def ConstOp: TensorflowMlrt_Op<"constop", []> { let summary = "The tf_mlrt ConstOp"; let description = [{ The ConstOp creates a constant tensorflow::Tensor from serialized proto. }]; let arguments = (ins StrAttr:$tensor_proto ); let results = (outs
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
// PadOp and CastOp. std::optional<BlockArgumentInfo> GetInputBlockArgNum(Value input) { auto block_arg_num = GetBlockArgNum(input); if (block_arg_num.has_value()) return block_arg_num; Value next_input = input; auto pad_op = dyn_cast_or_null<TF::PadOp>(next_input.getDefiningOp()); auto cast_op = dyn_cast_or_null<TF::CastOp>(next_input.getDefiningOp());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter.cc
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); if (devce_hardware == nullptr) return {};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 06:11:34 UTC 2024 - 7.5K 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/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/lite/experimental/tac/common/utils.h
namespace TFL { namespace tac { // Returns true if 'op' is non const op. Returns false otherwise or if // 'op' is null. inline bool IsNonConstOp(Operation* op) { if (!op) return false; if (llvm::isa<arith::ConstantOp, mlir::func::ConstantOp>(op)) return false; if (op->hasTrait<OpTrait::ConstantLike>()) return false; if (llvm::isa<TFL::ConstOp, TFL::QConstOp>(op)) return false; return true; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K 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/stablehlo/passes/prepare_quantize.cc
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/tensorflow/transforms/freeze_saved_model_assets.cc
ShapedType shaped_type = RankedTensorType::get({1}, TF::StringType::get(builder.getContext())); auto const_op = builder.create<TF::ConstOp>( asset.getLoc(), DenseStringElementsAttr::get(shaped_type, {filename})); for (auto init_op : init_table_from_text_file_ops_to_erase) { // Replace the InitializeTableFromTextFileV2Op to use the saved model's // asset filepath.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 30 01:12:09 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc
} else { // Convert I64 const array to I32. llvm::SmallVector<int32_t> const_i32_vec; for (auto element : const_value) { const_i32_vec.push_back(static_cast<int32_t>(element)); } const_value_raw = rewriter.getI32TensorAttr(const_i32_vec); } Value result_const = builder.create<TF::ConstOp>(const_value_raw); return result_const; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0)