- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 166 for setFAttr (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
// Name of the main function for the eventual executable needs to be set. main_func->setAttr(kTfSavedModelExportedNamesAttr, builder.getStrArrayAttr({func_name})); main_func->setAttr( kTfSavedModelInitializerTypeAttr, builder.getStringAttr(kTfSavedModelInitializerRestoreType)); module->setAttr("tf_saved_model.semantics", UnitAttr::get(builder.getContext()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc
dq.getArg()); dq.getResult().replaceAllUsesWith(dcast); if (auto extra_attr = op->getAttr(mlir::quant::kVolatileOpAttrName)) { dcast->setAttr(mlir::quant::kVolatileOpAttrName, extra_attr); } dq.erase(); } else if (auto q = llvm::dyn_cast<quantfork::QuantizeCastOp>(op)) { auto out_type = q.getResult().getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 02:50:01 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replica_id_to_device_ordinal.cc
// replicate_to_islands pass. Operation* op = require_device_ordinal_ops.front(); if (op->getAttrOfType<IntegerAttr>(kDeviceOrdinalAttr).getInt() == -1) { OpBuilder builder(op); op->setAttr(kDeviceOrdinalAttr, builder.getI64IntegerAttr(0)); } } else { // If the device ordinal attribute is -1, set it with the replica id // attribute modulo the number of TPU cores in the system.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4K 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/compiler/mlir/lite/experimental/tac/transforms/get_alternative_subgraph.cc
if (!interface_name.has_value()) { func.emitError("the func op does not have interface_name"); return nullptr; } cloned_func->setAttr( kDevice, builder->getStringAttr(target_device_inference_type.hardware)); cloned_func->setAttr(kInferenceType, builder->getStringAttr(GetInferenceString( target_device_inference_type.inference_type)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
auto new_key = StringAttr::get(ctx, old_key.str() + "_dtoh_0"); op->setAttr("key", new_key); } else if (isa<TF::XlaRecvFromHostOp>(op)) { StringRef old_key = op->getAttrOfType<StringAttr>("key").getValue(); auto new_key = StringAttr::get(ctx, old_key.str() + "_htod_0"); op->setAttr("key", new_key); } }); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
src/main/java/org/codelibs/fess/util/PrunedTag.java
Registered: Wed Jun 12 13:08:18 UTC 2024 - Last Modified: Thu Feb 22 01:37:57 UTC 2024 - 4.8K 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/decompose_reduce_dataset.cc
// Both the device attribute and compile_device_type attribute should be // propagated to the reduce function call. reduce_call->setAttr(kDeviceAttr, reduce_dataset->getAttrOfType<StringAttr>(kDeviceAttr)); reduce_call->setAttr( TF::kCompileDeviceTypeAttr, reduce_dataset->getAttrOfType<StringAttr>(TF::kCompileDeviceTypeAttr)); SmallVector<Value, 4> if_returns;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0)