Home
last modified time | relevance | path

Searched refs:input_arrays (Results 1 – 25 of 34) 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 …]
Dremove_trivial_passthrough.cc142 for (const auto& array : model->flags.input_arrays()) { in RemoveTrivialPassthroughOp()
/external/tensorflow/tensorflow/lite/python/
Dtflite_convert.py78 input_arrays = _parse_array(flags.input_arrays)
85 input_shapes = dict(zip(input_arrays, input_shapes_list))
89 "input_arrays": input_arrays,
135 input_arrays = converter.get_input_arrays()
145 if ((not flags.input_arrays and len(input_arrays) > 1) or
146 (len(input_arrays) != len(quant_stats))):
153 "values.".format(",".join(input_arrays)))
154 converter.quantized_input_stats = dict(zip(input_arrays, quant_stats))
231 if flags.graph_def_file and (not flags.input_arrays or
237 if not flags.input_arrays:
[all …]
Dlite.py69 def _run_graph_optimizations(graph_def, input_arrays, output_arrays, argument
87 for array in input_arrays + output_arrays:
475 input_arrays, argument
542 sess.graph, input_arrays)
551 if set(input_arrays) != set(input_shapes.keys()):
556 (name, input_shapes[name]) for name in input_arrays
568 input_arrays=None, argument
598 result = _freeze_saved_model(saved_model_dir, input_arrays, input_shapes,
606 input_arrays=None, argument
631 if input_arrays:
[all …]
Dconvert_saved_model_test.py162 input_arrays=None, argument
173 input_arrays=input_arrays,
221 self._convertSavedModel(saved_model_dir, input_arrays=["invalid-input"])
228 saved_model_dir, input_arrays=["Placeholder", "invalid-input"])
237 input_arrays=["Placeholder"],
249 input_arrays=["Placeholder"],
261 saved_model_dir=saved_model_dir, input_arrays=["inputB", "inputA"])
275 input_arrays=["inputA"],
284 saved_model_dir=saved_model_dir, input_arrays=["inputA"])
Dconvert_saved_model.py222 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
268 in_tensors = _get_tensors(graph, inputs, input_arrays)
Dconvert.py350 input_array = model.input_arrays.add()
405 input_array = model_flags.input_arrays.add()
Dlite_test.py962 saved_model_dir, input_arrays=['inputB', 'inputA'])
996 input_arrays=['inputA'],
1004 saved_model_dir, input_arrays=['inputA'], input_shapes={'inputA': None})
1097 keras_file, input_arrays=['invalid-input'])
1103 keras_file, input_arrays=['dense_input'])
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_lib.py141 def evaluate_frozen_graph(filename, input_arrays, output_arrays): argument
161 sess.graph, input_arrays)
254 input_arrays, argument
281 filename, input_arrays, output_arrays, input_shapes)
289 converter = _lite.TFLiteConverter.from_frozen_graph(filename, input_arrays,
319 input_arrays, argument
341 filename, input_arrays, output_arrays, input_shapes)
344 tf_eval_func = evaluate_frozen_graph(filename, input_arrays, output_arrays)
416 input_arrays=None, argument
436 filename, input_arrays=input_arrays, input_shapes=input_shapes)
/external/tensorflow/tensorflow/lite/python/testdata/
DBUILD15 "--input_arrays=input",
25 "--input_arrays=input",
40 "--input_arrays=input,indices",
/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.pbtxt15 …argspec: "args=[\'cls\', \'graph_def_file\', \'input_arrays\', \'output_arrays\', \'input_shapes\'…
19 …argspec: "args=[\'cls\', \'model_file\', \'input_arrays\', \'input_shapes\', \'output_arrays\'], v…
23 …argspec: "args=[\'cls\', \'saved_model_dir\', \'input_arrays\', \'input_shapes\', \'output_arrays\…
/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()
231 parsed_model_flags.input_arrays.specified() || in ReadModelFlagsFromCommandLineFlags()
247 if (parsed_model_flags.input_arrays.specified()) { in ReadModelFlagsFromCommandLineFlags()
250 absl::StrSplit(parsed_model_flags.input_arrays.value(), ',')) { in ReadModelFlagsFromCommandLineFlags()
307 if (model_flags->input_arrays().empty()) { in ReadModelFlagsFromCommandLineFlags()
Dmodel_cmdline_flags_test.cc55 EXPECT_EQ(result_flags.input_arrays.value(), expected_input_arrays); in TEST()
Dmodel_flags.proto137 repeated InputArray input_arrays = 1; field
164 // If true, will allow passing inexistent arrays in --input_arrays
170 // --input_arrays and --output_arrays. By default (if false), only
Dargs.h127 Arg<string> input_arrays; member
Dallocate_transient_arrays.cc240 for (const auto& input_array : model->flags.input_arrays()) { in AllocateTransientArrays()
/external/tensorflow/tensorflow/lite/g3doc/convert/
Dcmdline_examples.md50 --input_arrays=input \
69 contained within the SavedModel. The values for `--input_arrays` and
112 --input_arrays=input \
137 --input_arrays=input \
149 The flag `input_arrays` takes in a comma-separated list of input arrays as seen
160 …--input_arrays=InceptionV1/InceptionV1/Mixed_3b/Branch_1/Conv2d_0a_1x1/Relu,InceptionV1/InceptionV…
179 --input_arrays=input \
201 …--input_arrays=InceptionV1/InceptionV1/Mixed_3b/Branch_1/Conv2d_0a_1x1/Relu,InceptionV1/InceptionV…
215 When specifying intermediate arrays as `--input_arrays` and `--output_arrays`,
247 --input_arrays=input \
[all …]
Dquantization.md33 input_arrays = converter.get_input_arrays()
34 converter.quantized_input_stats = {input_arrays[0] : (0., 1.)} # mean, std_dev
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
Dcmdline_reference.md50 * `--input_arrays`. Type: comma-separated list of strings. Specifies the list
56 `--input_arrays` flag should correspond to each item in the following flags
66 * Example: `--input_arrays=foo,bar --input_shapes=2,3:4,5,6` means "foo"
/external/tensorflow/tensorflow/python/kernel_tests/
Dreduce_join_op_test.py204 input_arrays = [
211 self._testReduceJoin(input_arrays[i], truth, truth_shape, axis=i)
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_from_protos_test.py55 input_array = model_flags.input_arrays.add()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dimport.cc172 if (model_flags.input_arrays().empty()) { in ImportIOTensors()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_select.md74 --input_arrays=input \
86 --input_arrays=input \

12