Home
last modified time | relevance | path

Searched defs:exported_names (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/
Dpywrap_mlir.py50 def experimental_convert_saved_model_to_mlir(saved_model_path, exported_names, argument
58 exported_names, tags, argument
68 exported_names, tags, argument
Dmlir_wrapper.cc67 const std::string &exported_names, bool show_debug_info) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/
Dquantize_model.cc117 std::vector<std::string> exported_names = in QuantizeQatModel() local
212 std::vector<std::string> exported_names = in QuantizePtqModelPreCalibration() local
276 std::vector<std::string> exported_names = in QuantizePtqModelPostCalibration() local
359 std::vector<std::string> exported_names = in QuantizePtqDynamicRange() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dcommon.py39 def do_test(create_module_fn, exported_names=None, show_debug_info=False): argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_saved_model.cc98 auto exported_names = GetExportedNames(init_func_op); in verify() local
419 auto exported_names = in GetExportedNames() local
430 auto exported_names = in IsExported() local
510 auto exported_names = GetExportedNames(init_func_op); in GetSessionInitializerExportedName() local
/external/tensorflow/tensorflow/compiler/mlir/python/
Dmlir.cc234 std::vector<string> exported_names = in ExperimentalConvertSavedModelToMlir() local
254 std::vector<string> exported_names = in ExperimentalConvertSavedModelV1ToMlirLite() local
289 std::vector<string> exported_names = in ExperimentalConvertSavedModelV1ToMlir() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc155 absl::Span<std::string> exported_names, mlir::MLIRContext* context, in SavedModelObjectGraphToMlirImport()
179 absl::Span<std::string> exported_names, mlir::MLIRContext* context, in SavedModelSignatureDefsToMlirImport()
216 absl::Span<std::string> exported_names, mlir::MLIRContext* context, in SavedModelSignatureDefsToMlirImportLite()
Dimport_model.cc2837 absl::Span<std::string> exported_names) in ObjectNames()
3241 auto exported_names = mlir::tf_saved_model::GetExportedNames(global_tensor); in SortSavedModelModule() local
3259 auto exported_names = mlir::tf_saved_model::GetExportedNames(func); in SortSavedModelModule() local
3508 SavedModelV2Bundle* saved_model, absl::Span<std::string> exported_names, in Convert()
3703 std::optional<absl::Span<const std::string>> exported_names, in Convert()
3715 std::optional<absl::Span<const std::string>> exported_names, in SavedModelSignatureDefImporterLite()
4112 std::optional<absl::Span<const std::string>> exported_names, in Convert()
4257 absl::Span<std::string> exported_names, bool add_default_attributes, in ConvertSavedModelToMlir()
4265 const SavedModelBundle& saved_model, absl::Span<std::string> exported_names, in ConvertSavedModelV1ToMlir()
4279 std::optional<absl::Span<const std::string>> exported_names, in ConvertSavedModelV1ToMlirLite()
[all …]
/external/tensorflow/tensorflow/lite/experimental/mlir/testing/
Dmlir_convert.py193 exported_names=(), argument
/external/tensorflow/tensorflow/cc/experimental/libtf/mlir/
Dmlir_transform.cc43 absl::Span<std::string> exported_names(nullptr, 0); in LoadModule() local
/external/tensorflow/tensorflow/compiler/mlir/
Dtf_mlir_translate_main.cc120 absl::Span<std::string> exported_names(exported_names_vector); in main() local
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc133 absl::Span<std::string> exported_names(exported_names_in_vector); in ImportSavedModel() local
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc158 absl::Span<std::string> exported_names(exported_names_in_vector); in ConvertSavedModelToTFLiteFlatBuffer() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_tfl_translate.cc194 absl::Span<std::string> exported_names(exported_names_vector); in main() local
Dtf_to_tfl_flatbuffer.cc318 absl::Span<std::string> exported_names, const GraphImportConfig& specs, in ImportSavedModel()
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dinsert_main_function.cc68 auto exported_names = op->getAttrOfType<ArrayAttr>(kExportedNameAttr); in IsExported() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dlower_saved_model.cc471 if (auto exported_names = func_op->getAttrOfType<mlir::ArrayAttr>( in runOnOperation() local