- Sort Score
- Result 10 results
- Languages All
Results 161 - 170 of 242 for get_attr (0.09 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
ModuleOp module_op = op->getParentOfType<ModuleOp>(); // Ignore ops without quantization method. // Consider adding checks for individual methods. if (!op->getAttr(kQuantizationMethodAttr)) return failure(); // Ignore unquantized ops. if (!IsQuantizedXlaCallModuleOp(op)) return failure(); // For weight-only quantization, op should be hybrid quantized.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h
OperationState new_state(candidate_op->getLoc(), candidate_op->getName().getStringRef(), inputs, output_types, candidate_op->getAttrs()); for (int i = 0; i < candidate_op->getNumRegions(); ++i) { new_state.addRegion(); } Operation* quantized_op = rewriter.create(new_state); if (candidate_op->getNumRegions() != 0) {
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/tf2xla/transforms/legalize_tf_with_tf2xla.cc
encoding.getBounds()[i] == ShapedType::kDynamic) { return false; } } return true; } bool HasSymbolRefAttr(Operation* op) { for (const auto& attr : op->getAttrs()) { Attribute attr_value = attr.getValue(); if (mlir::isa<SymbolRefAttr>(attr_value)) { return true; } else if (auto array_attr = mlir::dyn_cast<ArrayAttr>(attr_value)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc
private: LogicalResult matchAndRewrite(TF::PartitionedCallOp call_op, PatternRewriter& rewriter) const override { StringRef function_name = mlir::cast<FlatSymbolRefAttr>(call_op.getFAttr()).getValue(); if (!function_name.starts_with("composite_") || !call_op->hasAttr(kQuantTraitAttrName)) { return failure(); } absl::Status check_status;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
} } OpBuilder builder(call); auto new_call = builder.create<CallOp>( call.getLoc(), info.decomposed_callee.getFunctionType().getResults(), new_operands, call->getAttrs()); new_call->setAttr( "f", SymbolRefAttr::get( builder.getContext(), const_cast<func::FuncOp&>(info.decomposed_callee).getName())); for (const auto& entry : info.ret_forward_input) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_dump_tensor_op.cc
new_call_op->setAttr(kEntryFuncAttrName, rewriter.getStringAttr(new_ref_func_name.getValue())); new_call_op->setAttrs(call_op->getAttrs()); new_call_op->removeAttr(rewriter.getStringAttr(kQuantTraitAttrName)); FlatSymbolRefAttr new_func_name_attr = FlatSymbolRefAttr::get(rewriter.getContext(), new_ref_func_name); new_call_op->setAttr(kEntryFuncAttrName, new_func_name_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 22:55:22 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc
new_result_types.push_back(while_op.getResultTypes()[i]); } } auto new_while_op = OpBuilder(while_op).create<WhileOp>( while_op.getLoc(), new_result_types, new_operands, while_op->getAttrs()); new_while_op.getCond().takeBody(while_op.getCond()); new_while_op.getBody().takeBody(while_op.getBody()); for (auto i = 0; i < n; ++i) { if (!while_op.getResult(i).use_empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
return failure(); } // Collect all the undefined attributes used in the inputs. llvm::SmallVector<StringAttr, 4> undefined_attrs; for (auto attr : input_used_attrs) { if (!func->getAttr(attr.getValue())) { undefined_attrs.push_back(attr); } } // Verify the argument order: tensors, tensor list, attributes; and also // verify there is at most one tensor list argument.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
new_operands.push_back(build_list_op.getOut()); } else { auto attr_name = compose_func.getArgAttrOfType<StringAttr>( arg.index(), kAttrArgumentNameAttr); auto attribute = op->getAttr(attr_name.getValue()); if (!attribute) { attribute = compose_func.getArgAttr(arg.index(), kAttrArgumentDefaultAttr); }
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/lite/transforms/post_quantize.cc
PatternRewriter& rewriter) const override { auto input_op = op.getInput().getDefiningOp(); if (auto q = llvm::dyn_cast_or_null<QuantizeOp>(input_op)) { if (!q->getAttr(mlir::quant::kVolatileOpAttrName)) return failure(); if (remove_volatile_ops_type == kPreserveInputsAndOutputs) { // Don't remove leading and trailing QDQ for PTQ workflow, so the io
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0)