- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 168 for setFAttr (0.14 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/utils/device_util_test.cc
mlir::MLIRContext context; mlir::OwningOpRef<mlir::ModuleOp> module_ref = mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); mlir::Builder builder(*module_ref); (*module_ref)->setAttr("tf.devices", builder.getBoolAttr(false)); mlir::TF::RuntimeDevices devices; EXPECT_TRUE(mlir::failed(GetDevicesFromOp(*module_ref, &devices))); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc
if (op->getDialect() != tf_dialect) return WalkResult::advance(); if (parallel_group_attr) { op->setAttr(TF::kParallelExecAnnotation, parallel_group_attr); } auto device_attr = op->getAttr(kDeviceAttr); if (!device_attr) { op->setAttr(kDeviceAttr, launch.getDeviceAttr()); return WalkResult::advance(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.h
if (reverse_permutation.empty()) return failure(); (*op)->setAttr("data_format", StringAttr::get(context, target_data_format)); for (auto pair : shuffle_attrs) { StringRef attr_name = pair.first; ArrayAttr attr_value = pair.second; (*op)->setAttr(attr_name, ShuffleArrayAttr(attr_value, reverse_permutation)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 5.3K bytes - Viewed (0) -
tensorflow/compat_template_v1.__init__.py
# Hook external TensorFlow modules. _current_module = _sys.modules[__name__] # Lazy load Keras v1 _tf_uses_legacy_keras = ( _os.environ.get("TF_USE_LEGACY_KERAS", None) in ("true", "True", "1")) setattr(_current_module, "keras", _KerasLazyLoader(globals(), mode="v1")) if _tf_uses_legacy_keras: _module_dir = _module_util.get_parent_dir_for_name("tf_keras.api._v1.keras") else:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 02:14:00 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/move_tpu_compile_to_front.cc
}; void MarkCompilationOps(Operation* func) { func->walk([&](Operation* op) { if (llvm::isa<TF::_TPUCompileMlirOp>(op)) { op->setAttr("_is_compilation", UnitAttr::get(func->getContext())); op = op->getParentOp(); while (op && op != func) { op->setAttr("_wraps_compilation", UnitAttr::get(func->getContext())); op = op->getParentOp(); } } }); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 17 00:26:18 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_cluster_formation.cc
// the function will have correct attributes. mlir::TF::CopyDeviceAndUnderscoredAttributes(call_op, cluster); cluster->setAttr(mlir::TF::kClusterOutlinedFunctionNameAttr, callee_name); cluster->setAttr(mlir::TF::kAllowSoftPlacementAttr, builder.getBoolAttr(true)); } // Encapsulate the first partitioned call that can be reached from
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 6K bytes - Viewed (0) -
docs/debugging/xattr/main.go
if err != nil { return 0, err } return binary.LittleEndian.Uint64(buf[:8]), nil } func listxattr(path string) ([]string, error) { return xattr.LList(path) } func setxattr(path, name string, value uint64) error { data := make([]byte, 8) binary.LittleEndian.PutUint64(data, value) return xattr.LSet(path, name, data) } func main() {
Registered: Sun Jun 16 00:44:34 UTC 2024 - Last Modified: Fri Dec 29 23:52:41 UTC 2023 - 3.2K 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/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)