- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 166 for setAttr (0.24 sec)
-
tensorflow/compat_template.__init__.py
_initializers = _KerasLazyLoader( globals(), submodule="initializers", name="initializers", mode="v2") setattr(_current_module, "losses", _losses) setattr(_current_module, "metrics", _metrics) setattr(_current_module, "optimizers", _optimizers)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 02:14:00 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize_compile_and_replicate_attributes.cc
// Convert `_tpu_replicate`. if (op->hasAttr(TF::kTpuReplicateAttr)) { op->setAttr(tensorflow::kReplicationInfoAttr, op->getAttr(TF::kTpuReplicateAttr)); op->removeAttr(tensorflow::kTpuReplicateAttr); op->setAttr(tensorflow::kCompileDeviceTypeAttr, builder.getStringAttr(tensorflow::kTpuDevice)); } // Convert `_XlaMustCompile`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_assignment.cc
if (device_attr.getValue().empty()) { op->setAttr(kDeviceAttr, builder.getStringAttr(default_device_)); } } else if (op->getDialect() == tf) { // Assign default device to all ops in Tensorflow dialect that do not // have device attribute. op->setAttr(kDeviceAttr, builder.getStringAttr(default_device_)); } }); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.3K 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/api_template.__init__.py
_initializers = _KerasLazyLoader( globals(), submodule="initializers", name="initializers") setattr(_current_module, "losses", _losses) setattr(_current_module, "metrics", _metrics) setattr(_current_module, "optimizers", _optimizers) setattr(_current_module, "initializers", _initializers) # Do an eager load for Keras' code so that any function/method that needs to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 06:27:59 UTC 2024 - 6.7K 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/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)