- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for lookupSymbolIn (0.41 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/visitor.cc
llvm::SmallDenseSet<func::FuncOp> added; for (const llvm::StringRef function_name : function_names) { auto func = llvm::dyn_cast_or_null<func::FuncOp>(symbol_table.lookupSymbolIn( module, builder.getStringAttr(function_name))); if (func == nullptr) { return module.emitError() << "Cannot find function '" << function_name << "'"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 03:46:51 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_session_initializer_to_function.cc
OpBuilder builder(session_initializer); const char *name = "session_initializer"; // In the (unlikely) case of there already being a session initializer // function, bail out. if (SymbolTable::lookupSymbolIn(module, name)) { module->emitWarning("session_initializer function already exists"); session_initializer.erase(); return; } auto init = builder.create<func::FuncOp>( module.getLoc(), name,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_inline_tpu_island.cc
Block &nested_block = nested_module->getRegion(0).front(); for (func::FuncOp func_op : llvm::make_early_inc_range(nested_block.getOps<func::FuncOp>())) { if (!symbol_table.lookupSymbolIn(getOperation(), func_op.getName())) { nested_block.getOperations().remove(func_op.getOperation()); symbol_table.insert(func_op.getOperation()); } } nested_module->erase(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 4K 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; }
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/tensorflow/ir/tf_ops_n_z.cc
<< padding.getShapedType().getShape(); } } auto module = op->getParentOfType<mlir::ModuleOp>(); auto func = dyn_cast_or_null<mlir::func::FuncOp>( SymbolTable::lookupSymbolIn(module, op.getComputation())); if (!func) { return op.emitOpError() << "has no reduction function specified"; } auto func_type = func.getFunctionType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
SymbolRefAttr func) { auto module = op->getParentOfType<ModuleOp>(); auto function = dyn_cast_or_null<func::FuncOp>(SymbolTable::lookupSymbolIn(module, func)); FunctionType function_ty = function.getFunctionType(); for (auto arg_in : llvm::zip(args, function_ty.getInputs())) { if (std::get<0>(arg_in).getType() != std::get<1>(arg_in)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
isa<TF::XlaVariadicSortOp>(op)) { auto propagate_shape_to = [&](mlir::SymbolRefAttr func_sym) { auto func = llvm::cast<mlir::func::FuncOp>( mlir::SymbolTable::lookupSymbolIn(module, func_sym)); mlir::SmallVector<mlir::Type, 2> types; for (auto type : func.getFunctionType().getInputs()) { types.push_back(tensorflow::GetTypeFromTFTensorShape(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)