- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for entry_func_1 (0.5 sec)
-
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/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/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) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec.cc
const auto module_op = call_op->getParentOfType<ModuleOp>(); const SymbolTable symbol_table(module_op); auto entry_func_op = dyn_cast_or_null<func::FuncOp>(symbol_table.lookup(function_name)); auto dot_general_op = *entry_func_op.getOps<DotGeneralOp>().begin(); if (auto optional_dim = GetDotGeneralQuantizationDim(dot_general_op); optional_dim) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 7.8K bytes - Viewed (0)