- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 50 for constop (0.17 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)