Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for input_arg_shapes_ (0.13 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.cc 

    // StableHLO module serialized into a XlaCallModuleOp. (ex: JAX/PyTorch models)
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
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 12:49:45 UTC 2024
    - 9.8K bytes
    - Viewed (0)
Back to top