- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 61 for hasAttr (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util.cc
return {"tf.entry_function", tf_saved_model::kTfSavedModelInitializerTypeAttr}; } bool IsEntryFunction(func::FuncOp func) { for (const auto &attr : GetEntryFunctionAttributeNames()) { if (func->hasAttr(attr)) { return true; } } return false; } llvm::SmallVector<func::FuncOp> GetEntryFunctions(ModuleOp module) { llvm::SmallVector<func::FuncOp> entry_funcs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
// Copy all referenced StableHLO functions to the new module. WalkResult result = WalkReachableFunctions( entry_func, [&](func::FuncOp f) -> WalkResult { if (!f->hasAttr(kFromXlaCallModuleAttrName)) { return WalkResult::advance(); } auto cloned = llvm::cast<func::FuncOp>(builder.clone(*f)); cloned->removeAttr(kFromXlaCallModuleAttrName);
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/lite/quantization/tensorflow/fallback_to_flex_patterns.td
// Marks the op as no fallback. def MarkNoFallback : NativeCodeCall<"SetNoFallbackAttr($_builder, $0)">; def NoFallbackAttrNotSet : Constraint<CPred< "!$0.getDefiningOp()->hasAttr(kNoFallbackAttr)">>; class FloatValueEquals<string val> : Constraint<CPred< "FloatValueEquals($0, " # val # ")">>; class RankEquals<string rank> : Constraint<CPred< "RankEquals($0, " # rank # ")">>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/restore_function_name.cc
void runOnOperation() override; }; void RestoreFunctionNameFromXlaCallModuleOp(TF::XlaCallModuleOp& call_op, SymbolTable& symbol_table) { if (!call_op->hasAttr(kOriginalStablehloEntryFunctionAttrName)) { return; } const auto original_function_name = call_op->getAttrOfType<StringAttr>( kOriginalStablehloEntryFunctionAttrName);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 08:32:43 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/order_by_dialect.cc
: info->DirectControlSuccessors(op); }; // Some tests have recursive calls and other shenanigans, so allow // them to skip side effect analysis. if (!func->hasAttr("ignore_side_effects_for_testing")) { info = &getAnalysis<mlir::TF::SideEffectAnalysis>().GetAnalysisForFunc(func); } func->walk([&](Operation* function) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 08 17:01:11 UTC 2022 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/rename_entrypoint_to_main.cc
// 'tf.entry_function' attribute set. llvm::SmallVector<func::FuncOp, 4> candidate_funcs; for (auto& entrypoint : entrypoints) { if (entrypoint.second->hasAttr("tf.entry_function")) { candidate_funcs.push_back(entrypoint.second); } } if (candidate_funcs.empty()) { return fail(module, "No entrypoints found"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_validate_inputs.cc
} } return success(); } LogicalResult HasTopLevelCompilationMarker( llvm::SmallVector<func::FuncOp> &entry_funcs) { for (auto &entry_func : entry_funcs) { if (entry_func->hasAttr(mlir::TF::kCompileDeviceTypeAttr)) { entry_func->emitError() << "TF2XLA MLIR Non-replicated Phase 1 Bridge " "does not support top-level compilation " "marker.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 19:29:14 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util_test.cc
} func.func @func(%arg0: tensor<i32>) -> tensor<i32> { func.return %arg0 : tensor<i32> } )mlir"; auto has_compile_device_type = [](mlir::SymbolUserOpInterface op) { return op->hasAttr(tensorflow::kCompileDeviceTypeAttr); }; mlir::MLIRContext context; context.loadDialect<mlir::func::FuncDialect, mlir::TF::TensorFlowDialect>(); mlir::OwningOpRef<mlir::ModuleOp> module =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.cc
//===----------------------------------------------------------------------===// mlir::LogicalResult IfrtCallOp::verify() { auto func = getOperation()->getParentOfType<mlir::func::FuncOp>(); if (func != nullptr && func->hasAttr("tfrt_ifrt_serving.program_id")) { return emitOpError() << "cannot be nested inside an IFRT program"; } for (mlir::Value arg : getArgs()) { if (mlir::isa<mlir::TF::ResourceType>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
private: LogicalResult matchAndRewrite(Operation* op, PatternRewriter& rewriter) const override { // Remove `_tpu_replicate` attributes on each operation first. if (op->hasAttr(tensorflow::kTPUReplicateAttr)) { op->removeAttr(tensorflow::kTPUReplicateAttr); return success(); } // Remove TPU operations. if (isa<TF::TPUReplicateMetadataOp, TF::TPUCompilationResultOp,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0)