Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for mlir_input (0.08 sec)

  1. tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_executor.h

      explicit RuntimeFallbackExecutor(int64_t num_threads);
    
      // Prepare() needs to be called once before calling Execute(). It sets up all
      // things necessary to execute the given 'mlir_input' with the fallback to
      // tensorflow.
      void Prepare(llvm::StringRef mlir_input);
    
      // Execute() can be called several times after the call to Prepare() (e.g. for
      // benchmarking).
      llvm::SmallVector<Tensor> Execute(llvm::StringRef function_name,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Nov 14 22:40:40 UTC 2022
    - 2.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_executor.cc

    }
    
    void RuntimeFallbackExecutor::Prepare(llvm::StringRef mlir_input) {
      // We only support IR written in the Tensorflow dialect.
      mlir::DialectRegistry registry;
      mlir::RegisterAllTensorFlowDialects(registry);
      mlir::MLIRContext context(registry);
    
      llvm::SourceMgr source_mgr;
      source_mgr.AddNewSourceBuffer(
          llvm::MemoryBuffer::getMemBuffer(mlir_input, "test_ir"), llvm::SMLoc());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 00:18:59 UTC 2024
    - 9.1K bytes
    - Viewed (0)
Back to top