- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for CreateRenameEntrypointToMainPass (0.59 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.cc
void AddTFToStablehloPasses( mlir::PassManager& pm, llvm::ArrayRef<llvm::ArrayRef<int64_t>> input_arg_shapes) { pm.addPass(mlir::odml::CreateRenameEntrypointToMainPass()); // TODO: b/230572023 - Consider improving shape inference for While op instead // of dropping the attribute. This need not be correct for models not trained // on TPU.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 9.8K bytes - Viewed (0)