Home
last modified time | relevance | path

Searched refs:input_arrays (Results 1 – 25 of 48) 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 (size_t 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.py123 input_arrays = _parse_array(flags.input_arrays)
130 input_shapes = dict(list(zip(input_arrays, input_shapes_list)))
134 "input_arrays": input_arrays,
184 input_arrays = converter.get_input_arrays()
194 if ((not flags.input_arrays and len(input_arrays) > 1) or
195 (len(input_arrays) != len(quant_stats))):
202 "values.".format(",".join(input_arrays)))
203 converter.quantized_input_stats = dict(list(zip(input_arrays, quant_stats)))
334 if flags.graph_def_file and (not flags.input_arrays or
340 if not flags.input_arrays:
[all …]
Dlite.py1890 input_arrays=None, argument
1916 if input_arrays or output_arrays:
1943 if input_arrays:
1944 input_tensors = _get_tensors_from_tensor_names(sess.graph, input_arrays)
2261 input_arrays, argument
2328 sess.graph, input_arrays)
2337 if set(input_arrays) != set(input_shapes.keys()):
2342 (name, input_shapes[name]) for name in input_arrays
2354 input_arrays=None, argument
2390 result = _freeze_saved_model(saved_model_dir, input_arrays, input_shapes,
[all …]
Dconvert_saved_model.py159 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
206 in_tensors = _get_tensors(graph, inputs, input_arrays)
/external/tensorflow/tensorflow/lite/testing/
Dtoco_convert.py32 input_arrays, argument
58 " --input_arrays=%s" % ",".join(input_arrays) +
101 input_arrays = [x[0] for x in input_tensors]
113 graphdef_file.name, input_arrays, output_tensors, input_shapes)
169 input_arrays=input_arrays,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc46 const std::vector<std::string>& input_arrays, in GraphdefToMlirImport() argument
69 TF_RETURN_IF_ERROR(ParseInputArrayInfo(input_arrays, input_dtypes, in GraphdefToMlirImport()
104 const std::vector<std::string>& input_arrays, in GraphdefToMlirTranslateFunction() argument
113 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToMlirTranslateFunction()
125 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToMlirTranslateFunction() argument
135 TF_RETURN_IF_ERROR(ParseNodeNames(input_arrays, input_array_vector)); in GraphdefToMlirTranslateFunction()
234 const std::vector<std::string>& input_arrays, in GraphdefToSplattedMlirTranslateFunction() argument
243 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToSplattedMlirTranslateFunction()
285 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToSplattedMlirTranslateFunction() argument
295 TF_RETURN_IF_ERROR(ParseNodeNames(input_arrays, input_array_vector)); in GraphdefToSplattedMlirTranslateFunction()
Dtf_mlir_translate.h43 const std::vector<std::string>& input_arrays,
60 absl::string_view input_arrays, absl::string_view input_dtypes,
71 const std::vector<std::string>& input_arrays,
87 absl::string_view input_arrays, absl::string_view input_dtypes,
Dtf_mlir_translate_registration.cc47 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToMlirTranslateFunction()
61 input, debug_info_file, input_arrays, input_dtypes, input_shapes, in GraphdefToSplattedMlirTranslateFunction()
Dtf_mlir_translate_cl.h30 extern llvm::cl::opt<std::string> input_arrays;
Dtf_mlir_translate_cl.cc26 opt<std::string> input_arrays( variable
/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.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\…
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc74 function_input_names.reserve(model_flags.input_arrays().size()); in HandleInputOutputArraysWithModule()
82 if (function_input_names_size != model_flags.input_arrays().size()) { in HandleInputOutputArraysWithModule()
85 ", expected: ", model_flags.input_arrays().size()); in HandleInputOutputArraysWithModule()
90 for (const auto& input_array : model_flags.input_arrays()) { in HandleInputOutputArraysWithModule()
172 if (!model_flags.input_arrays().empty() || in ConvertSavedModelToTFLiteFlatBuffer()
/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/lite/python/testdata/
DBUILD26 "--input_arrays=input",
36 "--input_arrays=input",
51 "--input_arrays=input,indices",
61 "--input_arrays=input,indices",
/external/tensorflow/tensorflow/lite/g3doc/r1/convert/
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 …]
/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()
313 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/tfjs/translate/
Dtf_to_tfjs_json.cc75 absl::string_view debug_info_file, absl::string_view input_arrays, in LoadFromGraphdefOrMlirSource() argument
95 file->getBuffer(), debug_info_file, input_arrays, input_dtypes, in LoadFromGraphdefOrMlirSource()
Dtf_to_tfjs_json.h41 absl::string_view debug_info_file, absl::string_view input_arrays,
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_to_tfl_flatbuffer.cc105 absl::string_view input_arrays, absl::string_view input_dtypes, in LoadFromGraphdefOrMlirSource() argument
128 file->getBuffer(), debug_info_file, input_arrays, input_dtypes, in LoadFromGraphdefOrMlirSource()
135 file->getBuffer(), debug_info_file, input_arrays, input_dtypes, in LoadFromGraphdefOrMlirSource()
Dtf_tfl_translate.cc195 specs, debug_info_file, input_arrays, input_dtypes, input_shapes, in main()
208 if (mlir::TFL::ParseInputNodeQuantSpecs(input_arrays, min_values, max_values, in main()
Dtf_to_tfl_flatbuffer.h44 absl::string_view input_arrays, absl::string_view input_dtypes,
/external/tensorflow/tensorflow/lite/experimental/mlir/testing/
Dmlir_convert.py49 input_arrays = [x[0] for x in input_tensors]
59 graphdef_file.name, input_arrays, output_tensors, input_shapes)

12