Home
last modified time | relevance | path

Searched refs:tag_set (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py75 for tag_set in sorted(tag_sets):
76 print('%r' % ', '.join(sorted(tag_set)))
79 def _show_signature_def_map_keys(saved_model_dir, tag_set): argument
91 signature_def_map = get_signature_def_map(saved_model_dir, tag_set)
134 def _show_inputs_outputs(saved_model_dir, tag_set, signature_def_key, indent=0): argument
148 tag_set)
297 for tag_set in sorted(tag_sets):
299 "contains the following SignatureDefs:" % ', '.join(tag_set))
301 tag_set = ','.join(tag_set)
302 signature_def_map = get_signature_def_map(saved_model_dir, tag_set)
[all …]
Dimport_pb_to_tensorboard.py43 def import_to_tensorboard(model_dir, log_dir, tag_set): argument
58 tag_set).graph_def
68 import_to_tensorboard(FLAGS.model_dir, FLAGS.log_dir, FLAGS.tag_set)
Dsaved_model_utils.py96 def get_meta_graph_def(saved_model_dir, tag_set): argument
117 set_of_tags = set([tag for tag in tag_set.split(",") if tag])
123 " found in SavedModel" % tag_set)
Dtools.bzl19 tag_set = "serve",
83 tag_set: The tag set to use in the SavedModel.
147 "--tag_set {} ".format(tag_set)
/external/llvm-project/llvm/cmake/modules/
DTensorFlowCompile.cmake2 # ${model} directory, looking for the ${tag_set} tag set, and the SignatureDef
7 function(tfcompile model tag_set signature_def_key fname cpp_class)
21 --tag_set ${tag_set}
/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model.py46 def get_meta_graph_def(saved_model_dir, tag_set): argument
60 return loader.load(sess, tag_set, saved_model_dir)
156 output_arrays, tag_set, signature_key): argument
187 meta_graph = get_meta_graph_def(saved_model_dir, tag_set)
Dconvert_saved_model_test.py87 tag_set=None, argument
89 if tag_set is None:
90 tag_set = set([tag_constants.SERVING])
99 tag_set=tag_set,
245 tag_set=set([saved_model.tag_constants.SERVING, "additional_test_tag"]))
Dlite.py1122 saved_model_dir, signature_key=signature_key, tag_set=tags)
1588 tag_set = set([_tag_constants.SERVING])
1590 result = _freeze_saved_model(temp_dir, None, None, None, tag_set,
1594 self._saved_model_tags = tag_set
1940 tag_set=None, argument
1963 if tag_set is None:
1964 tag_set = set([_tag_constants.SERVING])
1968 saved_model_converter = TFLiteSavedModelConverter(saved_model_dir, tag_set,
1974 output_arrays, tag_set, signature_key)
2060 tag_set=None, argument
[all …]
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_lib.py365 def evaluate_saved_model(directory, tag_set, signature_key): argument
378 if tag_set is None:
379 tag_set = set([_tag_constants.SERVING])
383 meta_graph = _loader.load(sess, tag_set, directory)
716 tag_set=None, argument
746 tag_set=tag_set,
757 tf_eval_func = evaluate_saved_model(directory, tag_set, signature_key)
767 tag_set=None, argument
790 model = _load.load(directory, tags=tag_set)
/external/tensorflow/tensorflow/compiler/mlir/python/
Dmlir.cc167 std::unordered_set<string> tag_set = in ExperimentalConvertSavedModelV1ToMlirLite() local
175 saved_model_path, tag_set, /*exported_names=*/{}, &context, in ExperimentalConvertSavedModelV1ToMlirLite()
191 std::unordered_set<string> tag_set = in ExperimentalConvertSavedModelV1ToMlir() local
196 tensorflow::LoadSavedModel({}, {}, saved_model_path, tag_set, &bundle); in ExperimentalConvertSavedModelV1ToMlir()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lite.-toco-converter.pbtxt18 …ed_model_dir\', \'input_arrays\', \'input_shapes\', \'output_arrays\', \'tag_set\', \'signature_ke…
Dtensorflow.lite.-t-f-lite-converter.pbtxt26 …ed_model_dir\', \'input_arrays\', \'input_shapes\', \'output_arrays\', \'tag_set\', \'signature_ke…
/external/libevent/
Devent_rpcgen.py1482 tag_set = None
1518 if not tag_set:
1519 tag_set = 1
1527 if not tag_set:
/external/llvm-project/polly/lib/External/ppcg/
Dgpu.c2242 isl_union_set *tag_set; in remove_local_accesses() local
2279 tag_set = isl_union_map_range(external); in remove_local_accesses()
2280 external = wrapped_reference_to_access(tag_set, tagged); in remove_local_accesses()
2284 tag_set = isl_union_map_domain(external); in remove_local_accesses()
2285 external = wrapped_reference_to_access(tag_set, tagged); in remove_local_accesses()
/external/tensorflow/tensorflow/c/
Dc_api.cc2216 std::unordered_set<string> tag_set; in TF_LoadSessionFromSavedModel()
2218 tag_set.insert(string(tags[i])); in TF_LoadSessionFromSavedModel()
2224 export_dir, tag_set, &bundle); in TF_LoadSessionFromSavedModel()
/external/libabigail/tests/data/test-diff-filter/
Dtest-PR27569-v1.abi10231 …<var-decl name="tag_set" type-id="cc26d15f" visibility="default" filepath="include/linux/blkdev.h"…
75234 …<var-decl name="tag_set" type-id="cc26d15f" visibility="default" filepath="include/linux/blkdev.h"…
Dtest-PR27569-v0.abi10231 …<var-decl name="tag_set" type-id="cc26d15f" visibility="default" filepath="include/linux/blkdev.h"…
75220 …<var-decl name="tag_set" type-id="cc26d15f" visibility="default" filepath="include/linux/blkdev.h"…