Home
last modified time | relevance | path

Searched refs:unconditionally_use_set_output_shapes (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dtf_mlir_translate.cc53 bool unconditionally_use_set_output_shapes, mlir::MLIRContext* context) { in GraphdefToMlirImport() argument
69 specs.unconditionally_use_set_output_shapes = in GraphdefToMlirImport()
70 unconditionally_use_set_output_shapes; in GraphdefToMlirImport()
113 bool unconditionally_use_set_output_shapes, mlir::MLIRContext* context) { in GraphdefToMlirTranslateFunction() argument
118 enable_shape_inference, unconditionally_use_set_output_shapes, context); in GraphdefToMlirTranslateFunction()
131 bool enable_shape_inference, bool unconditionally_use_set_output_shapes, in GraphdefToMlirTranslateFunction() argument
149 unconditionally_use_set_output_shapes, context); in GraphdefToMlirTranslateFunction()
156 bool unconditionally_use_set_output_shapes) { in SavedModelObjectGraphToMlirImport() argument
169 unconditionally_use_set_output_shapes); in SavedModelObjectGraphToMlirImport()
251 bool unconditionally_use_set_output_shapes, mlir::MLIRContext* context) { in GraphdefToSplattedMlirTranslateFunction() argument
[all …]
H A Dtf_mlir_translate.h52 bool enable_shape_inference, bool unconditionally_use_set_output_shapes,
67 bool enable_shape_inference, bool unconditionally_use_set_output_shapes,
83 bool enable_shape_inference, bool unconditionally_use_set_output_shapes,
99 bool enable_shape_inference, bool unconditionally_use_set_output_shapes,
109 bool unconditionally_use_set_output_shapes = false);
H A Dtf_mlir_translate_registration.cc52 enable_shape_inference, unconditionally_use_set_output_shapes, context); in GraphdefToMlirTranslateFunction()
66 enable_shape_inference, unconditionally_use_set_output_shapes, context); in GraphdefToSplattedMlirTranslateFunction()
H A Dimport_model.h72 bool unconditionally_use_set_output_shapes = false);
150 bool unconditionally_use_set_output_shapes = false);
H A Dimport_model.cc609 specs.unconditionally_use_set_output_shapes = in ConvertDeferredFunctions()
610 specs_.unconditionally_use_set_output_shapes; in ConvertDeferredFunctions()
948 if (specs_.unconditionally_use_set_output_shapes || in AddNodesToShapeRefiner()
1074 if (specs_.unconditionally_use_set_output_shapes) { in InferOutputType()
1160 if (specs_.unconditionally_use_set_output_shapes) { in InferOutputType()
2759 bool unconditionally_use_set_output_shapes);
3511 bool unconditionally_use_set_output_shapes) { in Convert() argument
3519 specs.unconditionally_use_set_output_shapes = in Convert()
3520 unconditionally_use_set_output_shapes; in Convert()
3705 bool unconditionally_use_set_output_shapes = false) { in Convert() argument
[all …]
H A Dmlir_import_options.h30 bool unconditionally_use_set_output_shapes = false; member
H A Dtf_mlir_translate_cl.h45 extern llvm::cl::opt<bool> unconditionally_use_set_output_shapes;
H A Dmlir_roundtrip_flags.h92 bool unconditionally_use_set_output_shapes = false; member
H A Dtf_mlir_translate_cl.cc121 opt<bool> unconditionally_use_set_output_shapes( variable
H A Dmlir_roundtrip_flags.cc64 << unconditionally_use_set_output_shapes; in str()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/
H A Dgraphdef_to_tfl_flatbuffer.cc90 specs.unconditionally_use_set_output_shapes = true; in ConvertGraphDefToTFLiteFlatBuffer()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dtf_to_tfl_flatbuffer.cc334 options.unconditionally_use_set_output_shapes = true; in ImportSavedModel()