- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for mlir_input (0.08 sec)
-
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) -
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)