- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 445 for FuncOp (0.11 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/pick_subgraphs.cc
// A subgraph may associate with 1...n FuncOp, and each FuncOp may correspond // with different hardwares. struct Subgraph { // The call can be thought as an "API". func::CallOp call; // available_choces can be viewed as "real implementation" assosicated with // the hardware. std::unordered_map<InferenceDeviceType, func::FuncOp, InferenceDeviceType::inference_device_type_hash>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 24 15:10:02 UTC 2022 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/cluster_tf_test.cc
TF_EXPECT_OK(RunFunctionTf2xlaClusteringBridge( *mlir_module_, /*is_supported_by_replicated_brige*/ true, /*is_in_fallback_enabled_mode=*/false)); FuncOp main = mlir_module_->lookupSymbol<mlir::func::FuncOp>("main"); ASSERT_TRUE(main); EXPECT_EQ(compilation_status.Delta(mlir::TF::kMlirPh1BridgeCounterReplicated, mlir::TF::kMlirPh1BridgeCounterV2,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/passes.td
def QuantizeWeightPass : Pass<"stablehlo-quantize-weight", "mlir::func::FuncOp"> { let summary = "Quantizes the weight component of StableHLO graph."; let dependentDialects = ["mlir::stablehlo::StablehloDialect"]; let constructor = "mlir::quant::stablehlo::CreateQuantizeWeightPass()"; } def UnfuseMhloBatchNormPass : Pass<"stablehlo-unfuse-mhlo-batch-norm", "mlir::func::FuncOp"> { let summary = "Unfuses batch normalization into arithmetic ops.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
void runOnOperation() override; }; // Move FuncOp referenced by `symbol_ref` from one symbol table to another. void MoveFuncOp(FlatSymbolRefAttr &symbol_ref, SymbolTable &from, SymbolTable &to) { if (to.lookup<func::FuncOp>(symbol_ref.getValue())) return; func::FuncOp callee = from.lookup<func::FuncOp>(symbol_ref.getValue()); callee.getOperation()->getBlock()->getOperations().remove(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
} return init_func_ops; } func::FuncOp GetInitializerFunction(ModuleOp module_op, const StringRef initializer_type) { SmallVector<func::FuncOp, 2> init_func_ops = GetInitializerFunctions(module_op); auto init_func_itr = absl::c_find_if( init_func_ops, [initializer_type](const func::FuncOp init_func_op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils.cc
StringRef(content.data(), content.size())); } inline TensorType GetInputType(func::FuncOp func, int idx) { return mlir::dyn_cast_or_null<TensorType>( func.getFunctionType().getInput(idx)); } inline TensorType GetResultType(func::FuncOp func, int idx) { return mlir::dyn_cast_or_null<TensorType>( func.getFunctionType().getResult(idx)); }
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/transforms/initialize_variables_in_session_init.cc
const_op.getResult()}); } func::FuncOp CreateSessionInitFunc(ModuleOp module) { constexpr char kSessionInitFuncName[] = "SessionInitializerFunction"; mlir::OpBuilder builder(module.getBodyRegion()); auto func_type = FunctionType::get(module.getContext(), /*inputs=*/{}, /*results=*/{}); auto func = builder.create<func::FuncOp>(module->getLoc(),
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/stablehlo/passes/bridge/passes.h
// qint types to int types and surround TF UniformQuantized ops with qint <-> // int casts. std::unique_ptr<OperationPass<func::FuncOp>> CreateConvertTFQuantTypesPass(); // Creates an instance of the VerifyQuantLegalization pass, which verifies all // quant ops and types are lowered. std::unique_ptr<OperationPass<func::FuncOp>> CreateVerifyQuantLegalizationPass(); // Add all passes for lowering TF quant ops and types to MHLO int.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 23 01:41:18 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
// Checks if the function is the main or initializer function. bool IsMainOrInitializerFunction(ModuleOp module, func::FuncOp func) { if (func.getSymName() == llvm::StringRef(tensorflow::kImportModelDefaultGraphFuncName) || func.getSymName() == kTfQuantSaveFuncName) { return true; } for (func::FuncOp init_func : tf_saved_model::GetInitializerFunctions(module)) {
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/utils/call_graph_util_test.cc
ASSERT_TRUE(result); EXPECT_EQ(noinline_pcall_ops.size(), 2); auto func = llvm::dyn_cast<mlir::func::FuncOp>(noinline_pcall_ops[0]->getParentOp()); ASSERT_TRUE(func); EXPECT_EQ(func.getSymName(), "while_body_func"); func = llvm::dyn_cast<mlir::func::FuncOp>(noinline_pcall_ops[1]->getParentOp()); ASSERT_TRUE(func); EXPECT_EQ(func.getSymName(), "outer_stateful_pcall_func");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 11.7K bytes - Viewed (0)