- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 103 for SetAttr (0.32 sec)
-
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/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) -
tensorflow/compiler/mlir/lite/debug/debug_test.cc
public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(MutatePass) void runOnOperation() override { mlir::OpBuilder builder(&getContext()); getOperation()->setAttr("tfl.random_attr", builder.getUnitAttr()); } }; class AlwaysFailPass : public mlir::PassWrapper<AlwaysFailPass, mlir::OperationPass<>> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(AlwaysFailPass)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 11:15:16 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
op_to_update->setAttr(kDeviceAttr, new_device_attr); PopulateDeviceForOpResults(*op_to_update, new_device_attr.getValue(), value_to_device); if (auto sink = llvm::dyn_cast<tf_executor::NextIterationSinkOp>(op_to_update)) { auto source = sink.GetSource(); source->setAttr(kDeviceAttr, new_device_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
builder.insert(metadata_op.clone()); for (Operation& op : func.getOps()) { if (!IsTPUOp(&op)) continue; op.setAttr(TF::kReplicationInfoAttr, new_group); } TF::TPUCompilationResultOp new_result = compilation_op.clone(); new_result->setAttr(kTpuCompilationStatus, new_group); builder.insert(new_result); } template <typename OpType> LogicalResult FindAndExcludeOp(func::FuncOp func,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
for (const auto& attr_name : attr_names) { if (!global_iter_id_op->hasAttr(attr_name)) continue; offset->setAttr(attr_name, global_iter_id_op->getAttr(attr_name)); new_global_iter_id->setAttr(attr_name, global_iter_id_op->getAttr(attr_name)); } // Make the next function to get inlined use a different offset. ++offset_value; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/analyze_variables.cc
getElementTypeOrSelf(operand.getType()))) { legalize_to_tfl = false; return WalkResult::interrupt(); } } return WalkResult::advance(); }); module->setAttr(kLegalizeTflVariables, BoolAttr::get(context, legalize_to_tfl)); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateAnalyzeVariablesPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
/*disabled_checks=*/empty_array_attr); xla_call_module_op->setAttr(TF::kStablehloEntryFunctionAttrName, SymbolRefAttr::get(stablehlo_func_op)); // Set jax.uses_shape_polymorphism=true to enable shape refinement at runtime. // This is needed for native serialization version >= 8. xla_call_module_op->setAttr( kStablehloModuleAttrsAttrName,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
// use it in exports or if there are breaking changes module->setAttr("tfl.schema_version", builder.getI32IntegerAttr(model->version)); if (!model->description.empty()) { module->setAttr("tfl.description", builder.getStringAttr(model->description)); } if (!metadata_attrs.empty()) { module->setAttr("tfl.metadata", builder.getDictionaryAttr(metadata_attrs)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
!HasOutsideCompiledAncestor(input_defining_op) && !input_defining_op->hasAttrOfType<StringAttr>( kXlaOutsideCompilationAttr)) { input_defining_op->setAttr( kXlaOutsideCompilationAttr, StringAttr::get(input_defining_op->getContext(), "auto")); outside_compiled_ops.push(input_defining_op); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0)