- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 168 for setFAttr (0.13 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/tac_filter.cc
FunctionFilter::FunctionFilterType type, OpBuilder& builder) { for (Operation& op : func.front()) { if (type == FunctionFilter::SKIP_TARGET_ANNOTATION) { op.setAttr(kSkipTargetAnnotation, builder.getUnitAttr()); } else if (type == FunctionFilter::INCLUDE_TARGET_ANNOTATION) { op.removeAttr(kSkipTargetAnnotation); } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_splits.cc
return WalkResult::advance(); for (Operation* pred : IslandPredecessors(split)) { if (auto colocation_classes = pred->getAttrOfType<ArrayAttr>(kClassAttr)) { split->setAttr(kClassAttr, colocation_classes); return WalkResult::advance(); } } } return WalkResult::advance(); }); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 18:44:34 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_to_legacy_compile_and_replicate_attributes.cc
auto result = func_op->walk([&](mlir::Operation* op) { if (failed(TF::HasValidCompilationAndReplicationAttributes(*op))) return WalkResult::interrupt(); if (op->hasAttr(TF::kReplicationInfoAttr)) { op->setAttr(TF::kTpuReplicateAttr, op->getAttr(TF::kReplicationInfoAttr)); op->removeAttr(TF::kReplicationInfoAttr); op->removeAttr(TF::kCompileDeviceTypeAttr); } return mlir::WalkResult::advance(); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
return ConstBytesAttr::get(builder->getContext(), StringRef(content.data(), content.size())); } } // namespace void ConvertNMSPaddedFunc::RewriteFunc() { func_->setAttr(kTFImplements, StringAttr::get(func_.getContext(), kTfNMSPadded)); Value boxes = func_.getArgument(0); Value scores = func_.getArgument(1); Value max_output_size = func_.getArgument(2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K 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/api_template_v1.__init__.py
import tensorflow_io_gcs_filesystem as _tensorflow_io_gcs_filesystem # 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")) _module_dir = _module_util.get_parent_dir_for_name("keras._tf_keras.keras") _current_module.__path__ = [_module_dir] + _current_module.__path__
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 02:14:00 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/get_arithmetic_count.cc
int64_t arithmetic_count = arithmetic_count_op.GetArithmeticCount(op); auto attr = builder.getIntegerAttr(builder.getIntegerType(64), arithmetic_count); op->setAttr("_arithmetic_count", attr); }); } } // namespace /// Creates an instance of the TensorFlow Lite dialect GetArithmeticCount /// pass.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 07 21:08:41 UTC 2022 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils.cc
} LogicalResult ConvertWhitespaceTokenizer(func::FuncOp func, llvm::StringRef api, FuncAttr attr) { func.eraseBody(); func.addEntryBlock(); func->setAttr(kTFImplements, attr); OpBuilder builder(func.getBody()); std::string empty_option_buffer; auto op = builder.create<CustomOp>( func.getLoc(), func.getFunctionType().getResults(), func.getArguments(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_launch_func_to_tf_call.cc
/*config=*/builder.getStringAttr(""), /*config_proto=*/builder.getStringAttr(""), /*executor_type=*/builder.getStringAttr("")); call_op->setAttr("device", launch->getAttrOfType<StringAttr>("device")); launch.replaceAllUsesWith(call_op); launch.erase(); }); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 21:08:09 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/translate_utils.cc
"bad_consumers", b.getI32ArrayAttr(llvm::ArrayRef<int32_t>( versions.bad_consumers().data(), versions.bad_consumers().data() + versions.bad_consumers().size()))); module->setAttr("tf.versions", b.getDictionaryAttr(llvm::ArrayRef<mlir::NamedAttribute>( {producer, min_consumer, bad_consumers}))); } mlir::LogicalResult ExtractTfVersions(mlir::ModuleOp module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 3.2K bytes - Viewed (0)