- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for entry_func_1 (0.23 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util_test.cc
ASSERT_TRUE(module); auto entry_funcs = GetEntryFunctions(*module); EXPECT_EQ(entry_funcs.size(), 2); EXPECT_EQ(entry_funcs[0].getSymName(), "entry_func_1"); EXPECT_EQ(entry_funcs[1].getSymName(), "entry_func_2"); } TEST(CallGraphUtilTest, GetCallees) { const char *const code = R"mlir( func.func @entry_func(%arg0: tensor<i32>) -> tensor<i32> attributes {tf_saved_model.initializer_type = ""} {
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/tf2xla/internal/mlir_bridge_pass_util_test.cc
EXPECT_TRUE( IsSupportedByReplicatedBridge(graph, /*function_library=*/nullptr)); } TEST(IsSupportedByReplicatedBridge, ReplicatedModule) { const char* const code = R"mlir( func.func @entry_func_1(%arg0: tensor<i32>) -> tensor<i32> attributes {tf.entry_function = {}} { %0 = "tf.Identity"(%arg0) {_tpu_replicate = "cluster"} : (tensor<i32>) -> (tensor<i32>) func.return %0 : tensor<i32> } )mlir";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_validate_inputs.cc
return failure(); } } 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 "
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/tf2xla/internal/passes/xla_cluster_formation.cc
SymbolTableCollection symbol_table_collection; SymbolTable symtab = symbol_table_collection.getSymbolTable(module); llvm::SmallVector<FuncOp> entry_funcs = GetEntryFunctions(module); for (auto &entry_func : entry_funcs) { if (EncapsulateFirstXlaCompilablePartitionedCalls( entry_func, symbol_table_collection, symtab) .failed()) { return signalPassFailure(); } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
// Returns `success()` if `entry_func_op`'s body is eligible for rewriting. At // this point `entry_func_op`'s signature has not been reset with quantized // types. virtual LogicalResult match(func::FuncOp entry_func_op, const Method& quantization_method) const = 0; // Rewrites the `entry_func_op`'s body. virtual void rewrite(func::FuncOp entry_func_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/xla_call_module_to_call.cc
SymbolTable symbol_table(module_op); auto entry_func_op = dyn_cast_or_null<func::FuncOp>( symbol_table.lookup(GetEntryFunctionName(op))); if (!entry_func_op) return failure(); // Replace the XlaCallModuleOp with a new CallOp. rewriter.replaceOpWithNewOp<func::CallOp>(op, entry_func_op, op.getArgs()); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 20:02:00 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util.cc
// have `tf_saved_model.initializer_type` instead. if (IsEntryFunction(func)) { entry_funcs.push_back(func); } }); return entry_funcs; } LogicalResult GetCallees(SymbolUserOpInterface op, SymbolTable &symtab, llvm::SmallVector<func::FuncOp> &callees) { for (auto attr : op->getAttrs()) { auto sym = mlir::dyn_cast<SymbolRefAttr>(attr.getValue());
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/translate/export_graphdef.cc
FunctionLibraryDefinition* flib_def, absl::flat_hash_set<Node*>* control_ret_nodes) { mlir::StringAttr entry_func_id = mlir::StringAttr::get(module.getContext(), "main"); std::optional<FuncOp> entry_func; FunctionLibraryDefinition temp_flib_def(OpRegistry::Global(), FunctionDefLibrary());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
FunctionLibraryDefinition* flib_def, absl::flat_hash_set<Node*>* control_ret_nodes) { mlir::StringAttr entry_func_id = mlir::StringAttr::get(module.getContext(), "main"); std::optional<FuncOp> entry_func; FunctionLibraryDefinition temp_flib_def(OpRegistry::Global(), FunctionDefLibrary());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
if (!entry_func_symbol) { return op.emitOpError() << "does not have " << kStablehloEntryFunctionAttrName << " attribute"; } auto entry_func = symbol_table.lookupSymbolIn<func::FuncOp>(module, entry_func_symbol); if (!entry_func) { return op.emitOpError() << "references an unknown entry function " << entry_func_symbol; } OpBuilder builder(module.getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0)