Searched defs:enable_shape_inference (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_mlir_translate.cc | 52 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToMlirImport()
110 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToMlirTranslateFunction()
129 bool enable_shape_inference, mlir::MLIRContext* context) { in GraphdefToMlirTranslateFunction()
224 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToSplattedMlirTranslateFunction()
273 bool enable_shape_inference, mlir::MLIRContext* context) { in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate_cl.cc | 114 opt<bool> enable_shape_inference( variable
|
| D | mlir_roundtrip_flags.h | 68 bool enable_shape_inference = true; member
|