Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dcommon.py43 def do_test(create_module_fn, exported_names=None, show_debug_info=False): argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_saved_model.cc96 auto exported_names = GetExportedNames(init_func_op); in Verify() local
417 auto exported_names = in GetExportedNames() local
428 auto exported_names = in IsExported() local
508 auto exported_names = GetExportedNames(init_func_op); in GetSessionInitializerExportedName() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc151 absl::Span<std::string> exported_names, mlir::MLIRContext* context) { in SavedModelObjectGraphToMlirImport()
171 absl::Span<std::string> exported_names, mlir::MLIRContext* context, in SavedModelSignatureDefsToMlirImport()
197 absl::Span<std::string> exported_names, mlir::MLIRContext* context, in SavedModelSignatureDefsToMlirImportLite()
Dimport_model.cc2599 absl::Span<std::string> exported_names) in ObjectNames()
3003 auto exported_names = mlir::tf_saved_model::GetExportedNames(global_tensor); in SortSavedModelModule() local
3021 auto exported_names = mlir::tf_saved_model::GetExportedNames(func); in SortSavedModelModule() local
3264 SavedModelV2Bundle* saved_model, absl::Span<std::string> exported_names, in Convert()
3458 SavedModelMLIRImportInput& input, absl::Span<std::string> exported_names, in Convert()
3473 absl::Span<std::string> exported_names, in SavedModelSignatureDefImporterLite()
3804 const SavedModelBundle& bundle, absl::Span<std::string> exported_names, in Convert()
3918 absl::Span<std::string> exported_names, bool add_default_attributes) { in ConvertSavedModelToMlir()
3924 const SavedModelBundle& saved_model, absl::Span<std::string> exported_names, in ConvertSavedModelV1ToMlir()
3932 absl::Span<std::string> exported_names, mlir::MLIRContext* context, in ConvertSavedModelV1ToMlirLite()
[all …]
/external/tensorflow/tensorflow/python/
Dpywrap_mlir.py41 def experimental_convert_saved_model_to_mlir(saved_model_path, exported_names, argument
Dmlir_wrapper.cc52 const std::string &exported_names, bool show_debug_info) { in PYBIND11_MODULE()
/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/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc151 absl::Span<std::string> exported_names(exported_names_in_vector); in ConvertSavedModelToTFLiteFlatBuffer() local
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc130 absl::Span<std::string> exported_names(exported_names_in_vector); in ImportSavedModel() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_tfl_translate.cc168 absl::Span<std::string> exported_names(exported_names_vector); in main() local
Dtf_to_tfl_flatbuffer.cc229 absl::Span<std::string> exported_names, const GraphImportConfig& specs, in ImportSavedModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dinitialize_variables.cc106 auto exported_names = attr.dyn_cast<ArrayAttr>(); in InitializeVariables() local
/external/tensorflow/tensorflow/compiler/mlir/python/
Dmlir.cc151 std::vector<string> exported_names = in ExperimentalConvertSavedModelToMlir() local