- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 167 for setFAttr (0.15 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/raise_target_subgraphs.cc
added_func_op->getRegion(0).getBlocks().front().front().getAttr( kInferenceType)); added_call_op->setAttr(kDevice, device); added_call_op->setAttr(kInferenceType, inference_type); added_func_op->setAttr(kDevice, device); added_func_op->setAttr(kInferenceType, inference_type); std::string function_name = absl::StrCat(interface_name.getValue().str(), "_",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K 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/tfr/utils/utils.cc
void AddAttributesInSameBlock(Block::iterator begin, Block::iterator end, const NamedAttrList& attrs) { for (Block::iterator it = begin; it != end; ++it) { for (auto& attr : attrs) { it->setAttr(attr.getName(), attr.getValue()); } } } // Adds `attrs` to all the operations between `begin` and `end`. Does not // include `end`. The operations might be across multiple blocks.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 6.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)