Home
last modified time | relevance | path

Searched refs:input_arrays (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_concatenation.cc36 void CopyTensorSegments(const std::vector<Array*>& input_arrays, in CopyTensorSegments() argument
40 for (Array* input_array : input_arrays) { in CopyTensorSegments()
52 CHECK(!input_arrays.empty()); in CopyTensorSegments()
55 input_arrays[0]->GetBuffer<A>().data.size() / array_copy_size[0]; in CopyTensorSegments()
59 src_ptr.reserve(input_arrays.size()); in CopyTensorSegments()
60 for (Array* input_array : input_arrays) { in CopyTensorSegments()
67 for (int i = 0; i < input_arrays.size(); i++) { in CopyTensorSegments()
79 void ConcatenateTensorBuffers(const std::vector<Array*>& input_arrays, in ConcatenateTensorBuffers() argument
88 std::vector<int> array_copy_size(input_arrays.size()); in ConcatenateTensorBuffers()
90 for (Array* input_array : input_arrays) { in ConcatenateTensorBuffers()
[all …]
/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model_test.py84 input_arrays=None, argument
96 input_arrays=input_arrays,
144 self._convertSavedModel(saved_model_dir, input_arrays=["invalid-input"])
151 saved_model_dir, input_arrays=["Placeholder", "invalid-input"])
160 input_arrays=["Placeholder"],
172 input_arrays=["Placeholder"],
184 saved_model_dir=saved_model_dir, input_arrays=["inputB", "inputA"])
198 input_arrays=["inputA"],
207 saved_model_dir=saved_model_dir, input_arrays=["inputA"])
Dtflite_convert.py85 input_arrays = _parse_array(flags.input_arrays)
92 input_shapes = dict(list(zip(input_arrays, input_shapes_list)))
96 "input_arrays": input_arrays,
142 input_arrays = converter.get_input_arrays()
152 if ((not flags.input_arrays and len(input_arrays) > 1) or
153 (len(input_arrays) != len(quant_stats))):
160 "values.".format(",".join(input_arrays)))
161 converter.quantized_input_stats = dict(list(zip(input_arrays, quant_stats)))
276 if flags.graph_def_file and (not flags.input_arrays or
282 if not flags.input_arrays:
[all …]
Dlite.py711 input_arrays, argument
778 sess.graph, input_arrays)
787 if set(input_arrays) != set(input_shapes.keys()):
792 (name, input_shapes[name]) for name in input_arrays
804 input_arrays=None, argument
834 result = _freeze_saved_model(saved_model_dir, input_arrays, input_shapes,
845 input_arrays=None, argument
869 if input_arrays or output_arrays:
897 if input_arrays:
898 input_tensors = _get_tensors_from_tensor_names(sess.graph, input_arrays)
[all …]
Dconvert_saved_model.py155 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
202 in_tensors = _get_tensors(graph, inputs, input_arrays)
Dutil.py180 input_arrays, argument
199 for array in input_arrays:
214 for array in input_arrays + output_arrays:
/external/tensorflow/tensorflow/lite/testing/
Dtoco_convert.py32 input_arrays, argument
60 " --input_arrays=%s" % ",".join(input_arrays) +
103 input_arrays = [x[0] for x in input_tensors]
116 graphdef_file.name, input_arrays, output_tensors, input_shapes)
162 input_arrays=input_arrays,
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_lib.py222 def evaluate_frozen_graph(filename, input_arrays, output_arrays): argument
250 inputs = _util.get_tensors_from_tensor_names(graph, input_arrays)
389 input_arrays, argument
416 filename, input_arrays, output_arrays, input_shapes)
424 converter = _lite.TFLiteConverter.from_frozen_graph(filename, input_arrays,
457 input_arrays, argument
489 filename, input_arrays, output_arrays, input_shapes)
492 tf_eval_func = evaluate_frozen_graph(filename, input_arrays, output_arrays)
621 input_arrays=None, argument
647 filename, input_arrays=input_arrays, input_shapes=input_shapes)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lite.-toco-converter.pbtxt10 …argspec: "args=[\'cls\', \'graph_def_file\', \'input_arrays\', \'output_arrays\', \'input_shapes\'…
14 …argspec: "args=[\'cls\', \'model_file\', \'input_arrays\', \'input_shapes\', \'output_arrays\'], v…
18 …argspec: "args=[\'cls\', \'saved_model_dir\', \'input_arrays\', \'input_shapes\', \'output_arrays\…
Dtensorflow.lite.-t-f-lite-converter.pbtxt16 …argspec: "args=[\'cls\', \'graph_def_file\', \'input_arrays\', \'output_arrays\', \'input_shapes\'…
20 …argspec: "args=[\'cls\', \'model_file\', \'input_arrays\', \'input_shapes\', \'output_arrays\', \'…
24 …argspec: "args=[\'cls\', \'saved_model_dir\', \'input_arrays\', \'input_shapes\', \'output_arrays\…
/external/tensorflow/tensorflow/lite/toco/logging/
Dconversion_log_util_test.cc143 InputArray input_arrays[3]; in TEST() local
144 input_arrays[0].set_name(input1); in TEST()
145 input_arrays[1].set_name(input2); in TEST()
146 input_arrays[2].set_name(input3); in TEST()
147 *model.flags.add_input_arrays() = input_arrays[0]; in TEST()
148 *model.flags.add_input_arrays() = input_arrays[1]; in TEST()
149 *model.flags.add_input_arrays() = input_arrays[2]; in TEST()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc48 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToMlirImport() argument
67 TF_RETURN_IF_ERROR(ParseInputArrayInfo(input_arrays, input_dtypes, in GraphdefToMlirImport()
102 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToMlirTranslateFunction() argument
108 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToMlirTranslateFunction()
163 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToSplattedMlirTranslateFunction() argument
169 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToSplattedMlirTranslateFunction()
Dtf_mlir_translate.h36 absl::string_view input_arrays, absl::string_view input_dtypes,
46 absl::string_view input_arrays, absl::string_view input_dtypes,
Dtf_mlir_translate_registration.cc46 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToMlirTranslateFunction()
57 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToSplattedMlirTranslateFunction()
Dtf_mlir_translate_cl.h29 extern llvm::cl::opt<std::string> input_arrays;
Dtf_mlir_translate_cl.cc25 opt<std::string> input_arrays( variable
/external/tensorflow/tensorflow/lite/python/testdata/
DBUILD16 "--input_arrays=input",
26 "--input_arrays=input",
41 "--input_arrays=input,indices",
/external/tensorflow/tensorflow/lite/toco/
Dmodel_cmdline_flags.cc49 Flag("input_arrays", parsed_flags.input_arrays.bind(), in ParseModelFlagsFromCommandLineFlags()
50 parsed_flags.input_arrays.default_value(), in ParseModelFlagsFromCommandLineFlags()
234 parsed_model_flags.input_arrays.specified() || in ReadModelFlagsFromCommandLineFlags()
250 if (parsed_model_flags.input_arrays.specified()) { in ReadModelFlagsFromCommandLineFlags()
253 absl::StrSplit(parsed_model_flags.input_arrays.value(), ',')) { in ReadModelFlagsFromCommandLineFlags()
310 if (model_flags->input_arrays().empty()) { in ReadModelFlagsFromCommandLineFlags()
Dmodel_cmdline_flags_test.cc54 EXPECT_EQ(result_flags.input_arrays.value(), expected_input_arrays); in TEST()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_to_tfl_flatbuffer.cc53 absl::string_view debug_info_file, absl::string_view input_arrays, in LoadFromGraphdefOrMlirSource() argument
88 file->getBuffer(), debug_info_file, input_arrays, input_dtypes, in LoadFromGraphdefOrMlirSource()
94 file->getBuffer(), debug_info_file, input_arrays, input_dtypes, in LoadFromGraphdefOrMlirSource()
Dtf_tfl_translate.cc136 debug_info_file, input_arrays, input_dtypes, input_shapes, in main()
148 if (mlir::TFL::ParseInputNodeQuantSpecs(input_arrays, min_values, max_values, in main()
Dtf_to_tfl_flatbuffer.h38 absl::string_view debug_info_file, absl::string_view input_arrays,
/external/tensorflow/tensorflow/lite/g3doc/r1/convert/
Dcmdline_examples.md48 --input_arrays=input \
67 contained within the SavedModel. The values for `--input_arrays` and
110 --input_arrays=input \
135 --input_arrays=input \
147 The flag `input_arrays` takes in a comma-separated list of input arrays as seen
158 …--input_arrays=InceptionV1/InceptionV1/Mixed_3b/Branch_1/Conv2d_0a_1x1/Relu,InceptionV1/InceptionV…
177 --input_arrays=input \
199 …--input_arrays=InceptionV1/InceptionV1/Mixed_3b/Branch_1/Conv2d_0a_1x1/Relu,InceptionV1/InceptionV…
213 When specifying intermediate arrays as `--input_arrays` and `--output_arrays`,
245 --input_arrays=input \
[all …]
Dpython_api.md74 input_arrays = ["input"]
78 graph_def_file, input_arrays, output_arrays)
97 `TFLiteConverter.from_saved_model()` are `input_arrays`, `input_shapes`,
174 input_arrays = converter.get_input_arrays()
175 converter.quantized_input_stats = {input_arrays[0] : (0., 1.)} # mean, std_dev
/external/tensorflow/tensorflow/lite/g3doc/convert/
Dquantization.md58 input_arrays = converter.get_input_arrays()
59 converter.quantized_input_stats = {input_arrays[0] : (0., 1.)} # mean, std_dev

12