Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for getLoadedDialect (0.13 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
Back to top