- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 445 for FuncOp (0.29 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/hoist_broadcast_read.cc
using mlir::LogicalResult; using mlir::Operation; using mlir::OperationPass; using mlir::OpOperand; using mlir::StringAttr; using mlir::success; using mlir::Value; using mlir::WalkResult; using mlir::func::FuncOp; using mlir::TF::ReadVariableOp; using mlir::tf_device::ReplicateOp; #define GEN_PASS_DEF_HOISTBROADCASTREADPASS #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args_test.cc
EXPECT_TRUE(module_op_ref); return module_op_ref; } mlir::MLIRContext ctx_{}; }; func::FuncOp GetMainFuncOp(ModuleOp module_op) { for (auto func_op : module_op.getOps<func::FuncOp>()) { if (func_op.getSymName() == "main") { return func_op; } } return {}; } TEST_F(ConvertAssetArgsTest, ConvertsSingleAssetArg) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
} pm.addNestedPass<mlir::func::FuncOp>(mlir::TF::CreateLowerQuantizedPass()); pm.addNestedPass<mlir::func::FuncOp>( mlir::quant::stablehlo::CreateConvertTFQuantTypesPass()); if (lower_to_xla_hlo) { for (auto& target_pass : custom_legalization_passes) { pm.addNestedPass<mlir::func::FuncOp>(std::move(target_pass)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/init_text_file_to_import_test_pass.cc
temp_file.os() << "grape"; temp_file.os().flush(); // Replace filename constant ops to use the temporary file. MLIRContext* context = &getContext(); for (func::FuncOp func : module.getOps<func::FuncOp>()) { llvm::SmallVector<arith::ConstantOp, 4> constant_ops( func.getOps<arith::ConstantOp>()); for (auto op : constant_ops) { ShapedType shaped_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 09:19:38 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h
// Returns the tf_saved_model.global_tensor op that func's arg_index'th argument // refers to as a bound input, or null. Operation *LookupBoundInput(func::FuncOp func, int arg_index, const SymbolTable &symbol_table); template <typename T> T LookupBoundInputOfType(func::FuncOp func, int arg_index, const SymbolTable &symbol_table) { return llvm::dyn_cast_or_null<T>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 18 03:21:34 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc
// ranges. bool SetInputNodesQuantizationParams(func::FuncOp func); // The function might contain more stats ops than required, and it will // introduce requantize if the calibration stats have conflicts. This method // tries to remove all the redundant stats ops. bool RemoveRedundantStats(func::FuncOp func); // Verify the quantization specification is expected for quantizing the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter_test.cc
} return absl::OkStatus(); } mlir::func::FuncOp GetMainFunc() { func::FuncOp main_func = module_->lookupSymbol<mlir::func::FuncOp>("main"); EXPECT_TRUE(main_func); return main_func; } mlir::Operation& GetFirstOpFromMain() { mlir::func::FuncOp main_func = GetMainFunc(); return main_func.getBody().front().front(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_device_inference.cc
auto module = getOperation(); const auto& resource_alias_analysis = getAnalysis<TF::ResourceAliasAnalysis>(); llvm::SmallDenseMap<func::FuncOp, PerFunctionResult, 4> per_function_results; llvm::SetVector<func::FuncOp> worklist; for (auto func_op : module.getOps<func::FuncOp>()) { worklist.insert(func_op); per_function_results.try_emplace( func_op, func_op, resource_alias_analysis.GetAnalysisForFunc(func_op));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 03:47:00 UTC 2023 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_duplicate_resource_ops.cc
constexpr StringRef kSharedNameAttr = "shared_name"; class MergeDuplicateResourceOpsPass : public PassWrapper<MergeDuplicateResourceOpsPass, OperationPass<func::FuncOp>> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(MergeDuplicateResourceOpsPass) StringRef getArgument() const final { return "quant-merge-duplicate-resource-ops"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 26 04:26:16 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/functional_control_flow_to_regions.cc
CopyDeviceAndUnderscoredAttributes(generator_op, generator_region); func::FuncOp init_function = SymbolTable::lookupNearestSymbolFrom<func::FuncOp>( generator_op, generator_op.getInitFunc()); func::FuncOp next_function = SymbolTable::lookupNearestSymbolFrom<func::FuncOp>( generator_op, generator_op.getNextFunc()); func::FuncOp finalize_function = SymbolTable::lookupNearestSymbolFrom<func::FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0)