- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for InlinerInterface (0.69 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_inline_tpu_island.cc
}; void ExecutorTPUV1IslandInliningPass::runOnOperation() { SymbolTable symbol_table(getOperation()); Operation *nested_module = symbol_table.lookup(kNestedModule); 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))
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/tfr/passes/decompose.cc
// rewrite will stop. return success(changed); } LogicalResult DecomposeTFOpsPass::InlineTFRFuncCalls() { // The Inliner will automatically use the registered dialect inliner. InlinerInterface inliner(&getContext()); func::FuncOp func = getOperation(); SymbolTable table(external_tfr_module_.has_value() ? *external_tfr_module_
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
#include "tensorflow/compiler/mlir/tfr/ir/tfr_types.h" namespace mlir { namespace TFR { //===----------------------------------------------------------------------===// // InlinerInterface //===----------------------------------------------------------------------===// namespace { /// This class defines the interface for inlining within the TFR dialect.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
op->setAttr(kTpuCompilationStatus, attr); } } // A helper class to inline TF::StatefulPartitionedCall ops struct Inliner : public InlinerInterface { Inliner(OpBuilder& builder, SymbolTable& symbol_table) : InlinerInterface(builder.getContext()), builder(builder), symbol_table(symbol_table) {} bool isLegalToInline(Operation* call, Operation* callable,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0)