Searched defs:enable_shape_inference (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | tf_mlir_translate.cc | 52 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToMlirImport()
112 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToMlirTranslateFunction()
131 bool enable_shape_inference, bool unconditionally_use_set_output_shapes, in GraphdefToMlirTranslateFunction()
250 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToSplattedMlirTranslateFunction()
300 bool enable_shape_inference, bool unconditionally_use_set_output_shapes, in GraphdefToSplattedMlirTranslateFunction()
|
| H A D | tf_mlir_translate_cl.cc | 115 opt<bool> enable_shape_inference( variable
|
| H A D | mlir_roundtrip_flags.h | 86 bool enable_shape_inference = true; member
|