Searched refs:enable_shape_inference (Results 1 – 9 of 9) 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() argument
68 specs.enable_shape_inference = enable_shape_inference; in GraphdefToMlirImport()
110 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToMlirTranslateFunction() argument
116 enable_shape_inference, context); in GraphdefToMlirTranslateFunction()
129 bool enable_shape_inference, mlir::MLIRContext* context) { in GraphdefToMlirTranslateFunction() argument
145 upgrade_legacy, enable_shape_inference, context); in GraphdefToMlirTranslateFunction()
224 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToSplattedMlirTranslateFunction() argument
230 enable_shape_inference, context); in GraphdefToSplattedMlirTranslateFunction()
273 bool enable_shape_inference, mlir::MLIRContext* context) { in GraphdefToSplattedMlirTranslateFunction() argument
289 upgrade_legacy, enable_shape_inference, context); in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate.h | 50 bool enable_shape_inference, mlir::MLIRContext* context);
64 bool enable_shape_inference, mlir::MLIRContext* context);
76 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference,
91 bool enable_shape_inference, mlir::MLIRContext* context);
|
| D | tf_mlir_translate_registration.cc | 50 enable_shape_inference, context); in GraphdefToMlirTranslateFunction()
64 enable_shape_inference, context); in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate_cl.h | 43 extern llvm::cl::opt<bool> enable_shape_inference;
|
| D | mlir_roundtrip_flags.h | 68 bool enable_shape_inference = true; member
|
| D | tf_mlir_translate_cl.cc | 114 opt<bool> enable_shape_inference( variable
|
| D | import_model.cc | 939 if (specs_.enable_shape_inference) { in InferInputType()
966 if (specs_.enable_shape_inference) { in InferOutputType()
1289 specs.enable_shape_inference = specs_.enable_shape_inference; in ConvertLibFunction()
1401 if (specs_.enable_shape_inference) { in PrepareConvert()
1413 if (!specs_.enable_shape_inference) { in PrepareConvert()
1456 if (!specs_.enable_shape_inference) { in Convert()
3907 specs.enable_shape_inference = false; in ConvertFunctionToMlir()
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass.cc | 201 import_config.enable_shape_inference = false; in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 668 config.enable_shape_inference = false; in GraphToModule()
|