- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 356 for setAttr (0.37 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/prepare_quantize.cc
if (!preceding_qcast) return failure(); auto new_qcast = rewriter.create<quantfork::QuantizeCastOp>( q_op.getLoc(), q_op.getType(), preceding_qcast.getArg()); new_qcast->setAttr(kVolatileOpAttrName, rewriter.getUnitAttr()); q_op->replaceAllUsesWith(new_qcast); return success(); } }; class ConvertTFConstOpToArithConstOp : public OpRewritePattern<TF::ConstOp> { public:
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/lite/python/jax_to_tfl_flatbuffer.cc
// Jax wrapped the output nodes in a tuple, so it's pretty hard to us // to tell the output at this point, we will set the output at the export // phase. main_func->setAttr("tf.entry_function", builder.getDictionaryAttr(attrs)); // StableHLO Quantizer is not supported for JAX input models, so // quantization_py_function_lib is set to nullptr.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dilated_conv.h
addtional_pad != bts_crops_attr.getValues<APInt>()[{i, j}].getSExtValue()) { op->setAttr("padding", rewriter.getStringAttr("SAME")); break; } } } // Set dilations op->setAttr("dilations", dilations_attr.value()); if (expand_op) { // If there is `expand_op`, we need to rewire the inputs to bypass the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/lift_quantizable_spots_as_functions.cc
if (!matcher.Match(xla_call_module_op)) continue; // Set the text representation of `Method` to matched // `TF::XlaCallModuleOp`. xla_call_module_op->setAttr( kQuantizationMethodAttr, StringAttr::get(module_op.getContext(), std::move(*quantization_method_txtpb))); } } return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h
template <typename Predicate> void CopyAttributes(Operation *from, Operation *to, Predicate P) { for (const NamedAttribute &attr : from->getAttrs()) if (P(attr)) to->setAttr(attr.getName(), attr.getValue()); } // Copies attributes whose name begins with an _ from `from` to `to`. inline void CopyUnderscoredAttributes(Operation *from, Operation *to) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
} if (cifg) { attributes.push_back(kCoupleInputForgetGates); } mlir::StringAttr attr_values = builder->getStringAttr(llvm::join(attributes, ",")); func->setAttr(kTFImplements, attr_values); return func; } class LstmUtilsTest : public ::testing::Test { protected: LstmUtilsTest() {} void SetUp() override { context_ = std::make_unique<mlir::MLIRContext>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
if (failed(::mlir::TF::UpdateDataFormat(data_format, this))) return failure(); // Update convolution attributes. (*this)->setAttr("dilations", ShuffleArrayAttr(getDilations(), perm)); (*this)->setAttr("strides", ShuffleArrayAttr(getStrides(), perm)); (*this)->setAttr("explicit_paddings", ShuffleArrayAttr(getExplicitPaddings(), perm, 2)); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
values.push_back(range.second - range.first); } auto attr_value = mlir::DenseI32ArrayAttr::get(tf_op->getContext(), values); tf_op->setAttr(attr_name, attr_value); }; if (tf_op->hasTrait<mlir::OpTrait::AttrSizedOperandSegments>() || tf_op->hasTrait<mlir::OpTrait::AttrSizedResultSegments>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/parallel_execute_to_islands.cc
group_annotation = absl::StrCat(parallel_group_attr.getValue().str(), ",", group_annotation); } for (auto& op : execute_island.GetBody()) { op.setAttr(TF::kParallelExecAnnotation, builder->getStringAttr(group_annotation)); } } executes.push_back(execute_island); } } void CreateIslandsFromParallelExecute(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 19 19:47:16 UTC 2023 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc
new_float_func.setType(FunctionType::get( getContext(), TypeRange{ValueRange{new_float_func_args}}, new_float_func.getResultTypes())); symbol_table.insert(new_float_func); op->setAttr("f", SymbolRefAttr::get(rewriter.getContext(), new_float_func.getName())); return success(); } LogicalResult matchAndRewrite(TF::PartitionedCallOp op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0)