- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TfrtFunctionCompileOptions (0.34 sec)
-
tensorflow/compiler/mlir/tfrt/function/function.h
namespace tfrt { class CoreRuntime; } namespace mlir { class ModuleOp; } namespace tensorflow { struct TfrtFunctionCompileOptions : public TfrtCompileOptions { // Currently only SavedModel API inference uses the tpu_fuse_ops option TfrtFunctionCompileOptions() { tpu_fuse_ops = false; // Currently grappler is not correctly applied in the eager execution of TF
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 04:50:20 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/function/function.cc
#include "tensorflow/core/platform/status.h" #include "tfrt/bef/bef_buffer.h" // from @tf_runtime #include "tfrt/bef_converter/mlir_to_bef.h" // from @tf_runtime namespace tensorflow { Status CompileTFMLIRToBEF(const TfrtFunctionCompileOptions& options, mlir::ModuleOp module, tfrt::BefBuffer* bef_buffer) { mlir::OpPrintingFlags print_flags; print_flags.elideLargeElementsAttrs(); if (VLOG_IS_ON(1)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 08:13:15 UTC 2024 - 3.8K bytes - Viewed (0)