- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for getLoadedDialect (0.13 sec)
-
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
RewritePatternSet patterns(ctx); populateWithGenerated(patterns); (void)applyPatternsAndFoldGreedily(func, std::move(patterns)); // Convert unsupported ops to Flex ops. auto tf_dialect = ctx->getLoadedDialect<TF::TensorFlowDialect>(); func.walk([&](Operation *op) { if (op->getDialect() != tf_dialect) return; if (IsAllowListedOp(op)) return; if (op->hasAttr(kNoFallbackAttr)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
FunctionLibraryDefinition temp_flib_def(OpRegistry::Global(), FunctionDefLibrary()); llvm::SmallDenseSet<FuncOp> visited_functions; auto tf_dialect = module.getContext()->getLoadedDialect("tf"); // Construct SymbolTable to enable cheap function lookups. The cost // of constructing the table is offset by the number of queries. SymbolTable symbol_table(module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
FunctionLibraryDefinition temp_flib_def(OpRegistry::Global(), FunctionDefLibrary()); llvm::SmallDenseSet<FuncOp> visited_functions; auto tf_dialect = module.getContext()->getLoadedDialect("tf"); // Construct SymbolTable to enable cheap function lookups. The cost // of constructing the table is offset by the number of queries. SymbolTable symbol_table(module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc
} } } island_op.erase(); return success(); } void ReplicateToIslandPass::runOnOperation() { const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); if (!tf_dialect) { getOperation().emitError() << "'tf' dialect is not registered"; return signalPassFailure(); } // Find islands with a single `tf_device.replicate` and create individual
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 24 21:01:40 UTC 2023 - 16.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
<< " outside compilation regions."; } } void MarkOpsForOutsideCompilation::runOnOperation() { auto module = getOperation(); const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); if (!tf_dialect) { getOperation().emitError() << "'tf' dialect is not registered"; return signalPassFailure(); } RewritePatternSet patterns(&getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0)