Searched refs:enable_shape_inference (Results 1 – 11 of 11) 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()
240 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToSplattedMlirTranslateFunction() argument
246 enable_shape_inference, context); in GraphdefToSplattedMlirTranslateFunction()
289 bool enable_shape_inference, mlir::MLIRContext* context) { in GraphdefToSplattedMlirTranslateFunction() argument
305 upgrade_legacy, enable_shape_inference, context); in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate.h | 51 bool enable_shape_inference, mlir::MLIRContext* context);
65 bool enable_shape_inference, mlir::MLIRContext* context);
77 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference,
92 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 | 44 extern llvm::cl::opt<bool> enable_shape_inference;
|
| D | mlir_roundtrip_flags.h | 76 bool enable_shape_inference = true; member
|
| D | tf_mlir_translate_cl.cc | 115 opt<bool> enable_shape_inference( variable
|
| D | mlir_roundtrip_flags.cc | 60 ss << "\nenable_shape_inference: " << enable_shape_inference; in str()
|
| D | import_model.cc | 605 specs.enable_shape_inference = specs_.enable_shape_inference; in ConvertDeferredFunctions()
1034 if (specs_.enable_shape_inference) { in InferInputType()
1061 if (specs_.enable_shape_inference) { in InferOutputType()
1472 if (specs_.enable_shape_inference) { in PrepareConvert()
1484 if (!specs_.enable_shape_inference) { in PrepareConvert()
1527 if (!specs_.enable_shape_inference) { in Convert()
3803 specs.enable_shape_inference = false; in ConvertGraph()
4110 specs.enable_shape_inference = false; in ConvertFunctionToMlir()
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass.cc | 204 import_config.enable_shape_inference = false; in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 704 config.enable_shape_inference = false; in GraphToModule()
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| D | saved_model.cc | 900 graph_import_config.enable_shape_inference = false; in ImportSubgraph()
|