Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_constant_concatenation.cc37 void CopyTensorSegments(const std::vector<Array*>& input_arrays, in CopyTensorSegments() argument
41 for (Array* input_array : input_arrays) { in CopyTensorSegments()
53 CHECK(!input_arrays.empty()); in CopyTensorSegments()
56 input_arrays[0]->GetBuffer<A>().data.size() / array_copy_size[0]; in CopyTensorSegments()
60 src_ptr.reserve(input_arrays.size()); in CopyTensorSegments()
61 for (Array* input_array : input_arrays) { in CopyTensorSegments()
68 for (size_t i = 0; i < input_arrays.size(); i++) { in CopyTensorSegments()
80 void ConcatenateTensorBuffers(const std::vector<Array*>& input_arrays, in ConcatenateTensorBuffers() argument
89 std::vector<int> array_copy_size(input_arrays.size()); in ConcatenateTensorBuffers()
91 for (Array* input_array : input_arrays) { in ConcatenateTensorBuffers()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dtflite_convert.py116 input_arrays = _parse_array(flags.input_arrays)
123 input_shapes = dict(list(zip(input_arrays, input_shapes_list)))
127 "input_arrays": input_arrays,
177 input_arrays = converter.get_input_arrays()
187 if ((not flags.input_arrays and len(input_arrays) > 1) or
188 (len(input_arrays) != len(quant_stats))):
195 "values.".format(",".join(input_arrays)))
196 converter.quantized_input_stats = dict(list(zip(input_arrays, quant_stats)))
326 if flags.graph_def_file and (not flags.input_arrays or
332 if not flags.input_arrays:
[all …]
H A Dconvert_saved_model_test.py80 input_arrays=None, argument
92 input_arrays=input_arrays,
140 self._convertSavedModel(saved_model_dir, input_arrays=["invalid-input"])
147 saved_model_dir, input_arrays=["Placeholder", "invalid-input"])
156 input_arrays=["Placeholder"],
168 input_arrays=["Placeholder"],
180 saved_model_dir=saved_model_dir, input_arrays=["inputB", "inputA"])
194 input_arrays=["inputA"],
203 saved_model_dir=saved_model_dir, input_arrays=["inputA"])
H A Dlite.py2241 input_arrays=None, argument
2267 if input_arrays or output_arrays:
2294 if input_arrays:
2295 input_tensors = _get_tensors_from_tensor_names(sess.graph, input_arrays)
2618 input_arrays, argument
2686 sess.graph, input_arrays)
2695 if set(input_arrays) != set(input_shapes.keys()):
2700 (name, input_shapes[name]) for name in input_arrays
2712 input_arrays=None, argument
2752 result = _freeze_saved_model(saved_model_dir, input_arrays, input_shapes,
[all …]
H A 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)
H A Dlite_test.py1261 input_arrays = quantized_converter.get_input_arrays()
1262 quantized_converter.quantized_input_stats = {input_arrays[0]: (0., 1.)}
1303 input_arrays = converter.get_input_arrays()
1304 converter.quantized_input_stats = {input_arrays[0]: (0., 1.)}
2107 saved_model_dir, input_arrays=['inputB', 'inputA'])
2178 input_arrays=['inputA'],
2188 input_arrays=['inputA'],
2197 saved_model_dir, input_arrays=['inputA'], input_shapes={'inputA': None})
2360 self._keras_file, input_arrays=['invalid-input'])
2366 self._keras_file, input_arrays=['dense_input'])
[all …]
H A Dconvert.py689 input_array = model_flags.input_arrays.add()
749 input_array = model_flags.input_arrays.add()
829 input_array = model_flags.input_arrays.add()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dtf_mlir_translate.cc46 const std::vector<std::string>& input_arrays, in GraphdefToMlirImport() argument
71 TF_RETURN_IF_ERROR(ParseInputArrayInfo(input_arrays, input_dtypes, in GraphdefToMlirImport()
106 const std::vector<std::string>& input_arrays, in GraphdefToMlirTranslateFunction() argument
115 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToMlirTranslateFunction()
127 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToMlirTranslateFunction() argument
138 TF_RETURN_IF_ERROR(ParseNodeNames(input_arrays, input_array_vector)); in GraphdefToMlirTranslateFunction()
244 const std::vector<std::string>& input_arrays, in GraphdefToSplattedMlirTranslateFunction() argument
253 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToSplattedMlirTranslateFunction()
296 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToSplattedMlirTranslateFunction() argument
307 TF_RETURN_IF_ERROR(ParseNodeNames(input_arrays, input_array_vector)); in GraphdefToSplattedMlirTranslateFunction()
H A Dtf_mlir_translate.h44 const std::vector<std::string>& input_arrays,
62 absl::string_view input_arrays, absl::string_view input_dtypes,
75 const std::vector<std::string>& input_arrays,
94 absl::string_view input_arrays, absl::string_view input_dtypes,
H A Dtf_mlir_translate_registration.cc49 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToMlirTranslateFunction()
63 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToSplattedMlirTranslateFunction()
/aosp_15_r20/external/tensorflow/tensorflow/lite/g3doc/r1/convert/
H A Dcmdline_examples.md78 --input_arrays=input \
84 and require additional flags `--input_arrays` and `--output_arrays` as this
100 --input_arrays=input \
128 --input_arrays=input \
153 --input_arrays=input \
165 --input_arrays=input \
172 The flag `input_arrays` takes in a comma-separated list of input arrays as seen
182 …--input_arrays=InceptionV1/InceptionV1/Mixed_3b/Branch_1/Conv2d_0a_1x1/Relu,InceptionV1/InceptionV…
197 --input_arrays=input \
216 …--input_arrays=InceptionV1/InceptionV1/Mixed_3b/Branch_1/Conv2d_0a_1x1/Relu,InceptionV1/InceptionV…
[all …]
H A Dcmdline_reference.md53 * `--input_arrays`. Type: comma-separated list of strings. Specifies the list
59 `--input_arrays` flag should correspond to each item in the following flags
68 * Example: `--input_arrays=foo,bar --input_shapes=2,3:4,5,6` means "foo"
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/logging/
H A Dconversion_log_util_test.cc144 InputArray input_arrays[3]; in TEST() local
145 input_arrays[0].set_name(input1); in TEST()
146 input_arrays[1].set_name(input2); in TEST()
147 input_arrays[2].set_name(input3); in TEST()
148 *model.flags.add_input_arrays() = input_arrays[0]; in TEST()
149 *model.flags.add_input_arrays() = input_arrays[1]; in TEST()
150 *model.flags.add_input_arrays() = input_arrays[2]; in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/
H A Dsaved_model_to_tfl_flatbuffer.cc77 function_input_names.reserve(model_flags.input_arrays().size()); in HandleInputOutputArraysWithModule()
86 if (function_input_names_size != model_flags.input_arrays().size()) { in HandleInputOutputArraysWithModule()
89 ", expected: ", model_flags.input_arrays().size()); in HandleInputOutputArraysWithModule()
94 for (const auto& input_array : model_flags.input_arrays()) { in HandleInputOutputArraysWithModule()
177 if (!model_flags.input_arrays().empty() || in ConvertSavedModelToTFLiteFlatBuffer()
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/testdata/
H A DBUILD26 "--input_arrays=input",
36 "--input_arrays=input",
51 "--input_arrays=input,indices",
61 "--input_arrays=input,indices",
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A 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()
313 if (model_flags->input_arrays().empty()) { in ReadModelFlagsFromCommandLineFlags()
H A Dmodel_flags.proto144 repeated InputArray input_arrays = 1; field
174 // If true, will allow passing inexistent arrays in --input_arrays
180 // --input_arrays and --output_arrays. By default (if false), only
H A Dmodel_cmdline_flags_test.cc54 EXPECT_EQ(result_flags.input_arrays.value(), expected_input_arrays); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A 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\…
H A Dtensorflow.lite.-t-f-lite-converter.pbtxt18 …argspec: "args=[\'cls\', \'graph_def_file\', \'input_arrays\', \'output_arrays\', \'input_shapes\'…
22 …argspec: "args=[\'cls\', \'model_file\', \'input_arrays\', \'input_shapes\', \'output_arrays\', \'…
26 …argspec: "args=[\'cls\', \'saved_model_dir\', \'input_arrays\', \'input_shapes\', \'output_arrays\…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dtf_to_tfl_flatbuffer.cc111 absl::string_view input_arrays, absl::string_view input_dtypes, in LoadFromGraphdefOrMlirSource() argument
135 file->getBuffer(), debug_info_file, input_arrays, input_dtypes, in LoadFromGraphdefOrMlirSource()
143 file->getBuffer(), debug_info_file, input_arrays, input_dtypes, in LoadFromGraphdefOrMlirSource()
H A Dtf_tfl_translate.cc226 specs, debug_info_file, input_arrays, input_dtypes, input_shapes, in main()
237 input_arrays, min_values, max_values, inference_type, &quant_specs)) { in main()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
H A Dtf_to_tfjs_json.cc76 absl::string_view debug_info_file, absl::string_view input_arrays, in LoadFromGraphdefOrMlirSource() argument
97 file->getBuffer(), debug_info_file, input_arrays, input_dtypes, in LoadFromGraphdefOrMlirSource()
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/
H A Dmlir_convert.py49 input_arrays = [x[0] for x in input_tensors]
59 graphdef_file.name, input_arrays, output_tensors, input_shapes)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/
H A Dreduce_join_op_test.py200 input_arrays = [
207 self._testReduceJoin(input_arrays[i], truth, truth_shape, axis=i)

12