- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 101 for constop (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
namespace mlir { namespace TF { namespace { // Creates ConstOp for int32_t value. ConstOp createI32ConstOp(int32_t value, Location loc, PatternRewriter* rewriter) { auto int_attr = IntegerAttr::get(rewriter->getIntegerType(32), value); return rewriter->create<ConstOp>(loc, int_attr); } // Creates ConstantOp for array of int32_t.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
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/lite/flatbuffer_import.cc
// TODO(krzysd) Handle function calls StatusOr<Operation*> ConvertOp( const tflite::OperatorT& op, const std::vector<Value>& vals_map, const std::vector<mlir::TensorType>& intermediate_types, Value optional_arg_marker, const std::vector<std::unique_ptr<tflite::OperatorCodeT>>& op_codes, const std::vector<std::string>& func_names, const std::vector<std::unique_ptr<tflite::TensorT>>& tensors, Location loc,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K 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)