- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for getCallableForCallee (0.29 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc
// // will be transformed to // // func.func @f(%x : bar) { // ... // } // ... // func.call @f(%x : bar) CallInterfaceCallable callable = call.getCallableForCallee(); mlir::SymbolRefAttr sym = callable.dyn_cast<mlir::SymbolRefAttr>(); auto symbol = mlir::SymbolTable::lookupNearestSymbolFrom(call.getOperation(), sym); if (!symbol) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
MutableOperandRange getArgOperandsMutable() { return getArgsMutable(); } // Return the callee of this operation. CallInterfaceCallable getCallableForCallee() { return getCalleeAttr(); } // Sets the callee from the callable void setCalleeFromCallable(CallInterfaceCallable callee); }]; let assemblyFormat = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td
// returns the function that this operation will launch. func::FuncOp getFuncOp() { return SymbolTable::lookupNearestSymbolFrom<func::FuncOp>(*this, getFuncAttr()); } CallInterfaceCallable getCallableForCallee() { return getFuncAttr(); } void setCalleeFromCallable(::mlir::CallInterfaceCallable callee); }]; } def TfDevice_RemoteRunOp : TfDevice_Op<"remote_run",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
} for (auto [ret, operand] : llvm::zip(call.getResults(), yield.getOperands())) { if (ret != operand) return std::nullopt; } SymbolRefAttr symbol = call.getCallableForCallee().get<SymbolRefAttr>(); if (!symbol) return std::nullopt; return symbol.getLeafReference(); } // Extracts the contents of a region with a single block into a new function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
StringAttr old_group) { // Adds the TPUReplicateMetatdataOp and TPUCompilationResultOp ops to the // function called by the provided 'caller'. mlir::CallInterfaceCallable callable = caller.getCallableForCallee(); mlir::SymbolRefAttr sym = callable.dyn_cast<mlir::SymbolRefAttr>(); auto func = llvm::dyn_cast<mlir::func::FuncOp>( mlir::SymbolTable::lookupNearestSymbolFrom(caller, sym));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0)