- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 537 for funcMap (0.12 sec)
-
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/tensorflow/transforms/sparsecore/embedding_program_key.cc
// tf_device.replicate. Operation* FindCompilePredecessor(Operation* func_op, Operation* preprocess_op) { llvm::DenseSet<Block*> blocks = GetAllBlocksBetween(preprocess_op, func_op); llvm::DenseMap<Block*, Operation*> scope; Operation* o = preprocess_op; while (o && o != func_op) { scope[o->getBlock()] = o; o = o->getParentOp(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_input_dialect_to_executor_pass.cc
return op->getName().getStringRef().str() == kClusterFuncOpName; } void VerifyInputDialectToExecutorPass::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { if (!tensorflow::tf2xla::internal::IsInBridgeAcceptableDialects(op)) { std::string error = "op is in dialect " +
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 08 16:32:56 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
// functions and avoid converting ConstOps inside initializer functions. for (auto func_op : module_op.getOps<func::FuncOp>()) { // Do not unfreeze constants under these functions. if (func_op.getSymName().contains("while_body")) continue; if (func_op.getSymName().contains("while_cond")) continue; absl::c_copy_if(func_op.getOps<TF::ConstOp>(), std::back_inserter(target_const_ops),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K 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/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/quantization/stablehlo/passes/remove_sharding_custom_call.cc
}; void RemoveShardingCustomCallPass::runOnOperation() { func::FuncOp func_op = getOperation(); MLIRContext& ctx = getContext(); RewritePatternSet patterns(&ctx); populateWithGenerated(patterns); FrozenRewritePatternSet frozen_patterns(std::move(patterns)); if (failed(applyPatternsAndFoldGreedily(func_op, frozen_patterns))) { func_op.emitWarning() << "Failed to converge "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 07:04:47 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
it->second.data_type; } } } func_op.eraseArguments(indices_to_erase); func_op.setType( FunctionType::get(func_op.getContext(), new_types, llvm::to_vector<4>(return_op->getOperandTypes()))); } // Lifts reads/writes of resource arguments from func_op and changes its // signature. resource_data_types is the (index, data type) pair for each
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K 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/tfrt/analysis/test_cost_analysis_pass.cc
llvm::StringRef getArgument() const final { return "tfrt-test-cost-analysis"; } llvm::StringRef getDescription() const final { return "Add remarks based on cost analysis for testing purpose."; } void runOnOperation() override { const auto& cost_analysis = getAnalysis<CostAnalysis>(); auto func_op = getOperation(); for (auto& op : func_op.front()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 23:46:29 UTC 2023 - 1.6K bytes - Viewed (0)