- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 21 for SetAttr (0.2 sec)
-
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
// function name can be changed by those passes. call_op->setAttr(TF::kStablehloEntryFunctionAttrName, FlatSymbolRefAttr::get(builder.getStringAttr(func_name))); // Store the custom attribute to restore the function name when loading it // back in the post calibration stage. As mentioned above, the above entry // function attribute is not reliable. call_op->setAttr(kOriginalStablehloEntryFunctionAttrName,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
return func.emitError() << "failed to rename StableHLO function " << func.getSymName(); } } if (is_main_func) { main_func_name = func.getSymNameAttr(); } func->setAttr(kFromXlaCallModuleAttrName, builder.getUnitAttr()); } if (!main_func_name) { return stablehlo_module.emitError() << "StableHLO module does not have an entry function"; } return main_func_name;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/statistics.cc
return; } const auto [min_value, max_value] = *min_max_values; mlir::OpBuilder builder(aggregator_op); aggregator_op->setAttr("min", builder.getF32FloatAttr(min_value)); aggregator_op->setAttr("max", builder.getF32FloatAttr(max_value)); }); return status; } bool IsCalibrationRequired(mlir::ModuleOp module_op) { bool calibration_required = false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/views/attr_view.cc
default_value); } string AttrView::SetterMethod() const { if (!attr_.is_list()) { return absl::StrCat("SetAttr", toUpperCamel(attr_.full_type())); } else { return absl::StrCat("SetAttr", toUpperCamel(attr_.base_type()), "List"); } } std::vector<string> AttrView::SetterArgs() const { if (attr_.full_type() == "string") {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 03 07:02:00 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
ConstOp zero = builder.create<ConstOp>(loc, zero_attr); zero->setAttr(kICIWeightDistributionMlirBridgeMarker, builder.getBoolAttr(true)); ConstOp shape = builder.create<ConstOp>(loc, shape_attr); shape->setAttr(kICIWeightDistributionMlirBridgeMarker, builder.getBoolAttr(true)); FillOp fill = builder.create<FillOp>(loc, shape, zero); fill->setAttr(kICIWeightDistributionMlirBridgeMarker,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
auto func = builder.create<func::FuncOp>(module->getLoc(), kSessionInitFuncName, func_type); func->setAttr(kTfSavedModelExportedNamesAttr, builder.getStrArrayAttr({kSessionInitFuncName})); func->setAttr(kTfSavedModelInitializerTypeAttr, builder.getStringAttr(kTfSavedModelInitializerRestoreType)); func.setVisibility(mlir::func::FuncOp::Visibility::Public);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
} results->push_back(value.getInt()); } return success(); } } // namespace LogicalResult ConvertMaxUnpoolingFunc::RewriteFunc() { func_.eraseBody(); func_.addEntryBlock(); func_->setAttr(kTFImplements, StringAttr::get(func_.getContext(), kMaxUnpooling)); OpBuilder builder(func_.getBody()); std::string custom_option_buffer; if (failed(CreateCustomOptions(custom_option_buffer))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
// use it in exports or if there are breaking changes module->setAttr("tfl.schema_version", builder.getI32IntegerAttr(model->version)); if (!model->description.empty()) { module->setAttr("tfl.description", builder.getStringAttr(model->description)); } if (!metadata_attrs.empty()) { module->setAttr("tfl.metadata", builder.getDictionaryAttr(metadata_attrs)); }
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/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc
OpBuilder builder(&ctx); module_op.walk([&builder](Operation* op) { if (op->hasAttrOfType<BoolAttr>("is_stateless") && ContainCalibrationStatisticsSaverOp(op)) { op->setAttr("is_stateless", builder.getBoolAttr(false)); } }); } std::unique_ptr<OperationPass<ModuleOp>> CreateInsertCalibrationStatisticsSaverPass( StringRef calibration_data_dir,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
getContext(), ftype.getResults(), graph_func.getAllResultAttrs(), res_types, res_attrs))) return failure(); // Update the function type which has excluded the control args. func->setAttr("function_type", TypeAttr::get(rewriter.getFunctionType( arg_types, res_types))); // Update arg/result attributes. func.setAllArgAttrs(arg_attrs);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0)