- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 81 for func_ops (1.09 sec)
-
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
return false; } SmallVector<func::FuncOp> GetSortedFunctions(ModuleOp module_op) { auto iterator_range = module_op.getOps<func::FuncOp>(); SmallVector<func::FuncOp> func_ops(iterator_range.begin(), iterator_range.end()); absl::c_sort(func_ops, [](func::FuncOp op1, func::FuncOp op2) { return op1.getName() < op2.getName(); }); return func_ops; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc
// Insert CalibrationStatisticsSaverOp to the end of each region. for (auto func_op : module_op.getOps<func::FuncOp>()) { int32_t output_file_idx = 0; StringRef func_name = func_op.getSymName(); func_op.walk([&output_file_idx, &ctx, &func_name, &aggregator_ops_to_ignore, this](Operation* op) { for (Region& region : op->getRegions()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
return failure(); } // Fetch all VarHandleOp. llvm::StringSet<> variable_names; llvm::SmallVector<TF::VarHandleOp, 4> var_ops; for (auto func_op : module.getOps<func::FuncOp>()) { for (auto var_handle_op : func_op.getOps<TF::VarHandleOp>()) { auto variable_name = GetVariableName(var_handle_op); if (variable_names.count(variable_name)) continue; var_ops.emplace_back(var_handle_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints_test.cc
OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kArgumentWithUnknownDims); ASSERT_TRUE(module_op); auto func_op = module_op->lookupSymbol<func::FuncOp>("unknown_dims_arg"); ASSERT_THAT(func_op, NotNull()); ASSERT_THAT(func_op.getNumArguments(), Eq(1)); EXPECT_TRUE(HasRankOf(func_op.getArgument(0), /*rank=*/2)); } TEST_F(AttrsAndConstraintsTest, HasRankOfReturnsFalseForUnknownRank) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 22.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
} return success(); } void LiftHashTableOpsAsArgsPass::runOnOperation() { auto module_op = getOperation(); for (auto func_op : module_op.getOps<func::FuncOp>()) { if (failed(LiftHashTableOpsToArguments(module_op, func_op))) { signalPassFailure(); return; } } } static PassRegistration<LiftHashTableOpsAsArgsPass> pass; } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc
// we do not overwrite the function argument mapping here. func_op.walk([&](Operation* op) { if (op == func_op) return; for (Region& region : op->getRegions()) { assign_unique_id_to_all(region.getArguments()); } }); llvm::SmallDenseMap<ResourceHandle, int64_t> resource_handle_id_map; func_op.walk([&](Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.cc
return false; // not a call func::FuncOp func_op = dyn_cast<func::FuncOp>(call.resolveCallable( &symbol_table_collection_)); return IsPureFunction(func_op); } bool OpSideEffectCollector::IsPureFunction(func::FuncOp func_op) const { auto it = is_pure_function_.find(func_op); if (it == is_pure_function_.end()) { bool is_pure = true; is_pure_function_[func_op] = is_pure; // prevent infinite recursion
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 41.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
} }; // Returns true iff func_op has either no Region or the body has no Blocks. bool IsFuncOpEmpty(func::FuncOp func_op) { return func_op->getNumRegions() == 0 || func_op.getBody().empty(); } // Gets the GraphOp from the function op. Returns an empty op iff it doesn't // exist. GraphOp GetGraphOpFromFuncOp(func::FuncOp func_op) { if (IsFuncOpEmpty(func_op)) return {};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
for (func::FuncOp func_op : getOperation().getOps<func::FuncOp>()) { if (functions_to_skip.contains(func_op)) { OpBuilder builder(func_op); // Mark this function as being skipped in island outlining. func_op->setAttr(mlir::TF::kSkipIslandOutlining, builder.getBoolAttr(true)); continue; } func_op.walk([&](GraphOp graph) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.h
return alias_analysis_; } private: // Runs the analysis and populates `sorted_control_predecessors_` and // `sorted_control_successors_` for `func_op`. Clears `control_predecessors_`. void AnalyzeFunction(func::FuncOp func_op); // Runs the analysis and populates `control_predecessors_` for `region`. void AnalyzeRegion(Region* region);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 14.8K bytes - Viewed (0)