Home
last modified time | relevance | path

Searched refs:TensorFlowSavedModelDialect (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_saved_model.cc125 TensorFlowSavedModelDialect::TensorFlowSavedModelDialect(MLIRContext *context) in TensorFlowSavedModelDialect() function in mlir::tf_saved_model::TensorFlowSavedModelDialect
127 TypeID::get<TensorFlowSavedModelDialect>()) { in TensorFlowSavedModelDialect()
189 LogicalResult TensorFlowSavedModelDialect::verifyRegionArgAttribute( in verifyRegionArgAttribute()
217 LogicalResult TensorFlowSavedModelDialect::verifyRegionResultAttribute( in verifyRegionResultAttribute()
245 ModuleOp module, TensorFlowSavedModelDialect *dialect) { in VerifySavedModelModule()
377 LogicalResult TensorFlowSavedModelDialect::verifyOperationAttribute( in verifyOperationAttribute()
H A Dtf_saved_model.h27 class TensorFlowSavedModelDialect : public Dialect {
29 explicit TensorFlowSavedModelDialect(MLIRContext *context);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/
H A Dquantize_model.cc130 mlir::tf_saved_model::TensorFlowSavedModelDialect, in QuantizeQatModel()
218 mlir::tf_saved_model::TensorFlowSavedModelDialect, in QuantizePtqModelPreCalibration()
289 mlir::tf_saved_model::TensorFlowSavedModelDialect, in QuantizePtqModelPostCalibration()
372 mlir::tf_saved_model::TensorFlowSavedModelDialect, in QuantizePtqDynamicRange()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/
H A Ddialect_registration.h41 mlir::tf_saved_model::TensorFlowSavedModelDialect, in RegisterAllTensorFlowDialects()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dtf_quant_opt.cc36 mlir::tf_saved_model::TensorFlowSavedModelDialect, in main()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dlower_saved_model.cc505 if (llvm::isa<mlir::tf_saved_model::TensorFlowSavedModelDialect>( in runOnOperation()