- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for getRootReference (0.4 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util.cc
auto sym = mlir::dyn_cast<SymbolRefAttr>(attr.getValue()); if (!sym) continue; auto callee = symtab.lookup<func::FuncOp>(sym.getRootReference()); if (!callee) { // This is not expected to happen in practice. return op->emitError() << "Cannot find function " << sym.getRootReference(); } callees.push_back(callee); } return success(); } bool HasSingleBlock(func::FuncOp func) {
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/executor_tpuv1_inline_tpu_island.cc
if (!nested_module) return; InlinerInterface inliner(&getContext()); auto walk_result = getOperation().walk([&](TF::PartitionedCallOp call_op) { if (!call_op.getF().getRootReference().getValue().starts_with( kNestedModule)) return WalkResult::advance(); // This is a call we need to inline! LLVM_DEBUG(llvm::dbgs()
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/tf2xla/internal/passes/xla_cluster_formation.cc
CallInterfaceCallable callable = call.getCallableForCallee(); auto sym = callable.get<mlir::SymbolRefAttr>(); EncapsulatePartitionedCall(pcall_op, sym.getRootReference()); } // Partitioned calls are executed asynchronous. The calls outside of // device clusters therefore should not be inlined to perserve run-time // performance.
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/tensorflow/ir/tf_device.cc
//===----------------------------------------------------------------------===// LogicalResult ClusterFuncOp::verifySymbolUses( mlir::SymbolTableCollection& symbolTable) { StringAttr func_attr = getFuncAttr().getRootReference(); func::FuncOp func = symbolTable.lookupNearestSymbolFrom<func::FuncOp>(*this, func_attr); if (!func) { return emitError("'func' attribute refers to an undefined function: ")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
//===----------------------------------------------------------------------===// LogicalResult BatchFunctionOp::verifySymbolUses( SymbolTableCollection& symbolTable) { StringAttr func_attr = getFAttr().getRootReference(); func::FuncOp func = symbolTable.lookupNearestSymbolFrom<func::FuncOp>(*this, func_attr); if (!func) { return emitError("'f' attribute refers to an undefined function: ")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
return false; } ModuleOp module = op->getParentOfType<ModuleOp>(); func::FuncOp func = module.lookupSymbol<func::FuncOp>( op.getShapeInferenceGraphAttr().getRootReference()); if (func == nullptr) return false; std::vector<_XlaRecvAtHostOp> xla_recv_at_host_ops; func.walk([&](_XlaRecvAtHostOp op) { xla_recv_at_host_ops.push_back(op); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)