Home
last modified time | relevance | path

Searched refs:output_arrays (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/lite/python/
Dlite.py69 def _run_graph_optimizations(graph_def, input_arrays, output_arrays, argument
87 for array in input_arrays + output_arrays:
408 output_arrays=None): argument
449 if not input_arrays_with_shape or not output_arrays:
454 self._output_arrays = output_arrays
476 output_arrays, argument
544 sess.graph, output_arrays)
563 output_arrays=output_arrays)
570 output_arrays=None, argument
599 output_arrays, tag_set, signature_key)
[all …]
Dconvert_saved_model_test.py164 output_arrays=None, argument
175 output_arrays=output_arrays,
211 self._convertSavedModel(saved_model_dir, output_arrays=["invalid-output"])
238 output_arrays=["add"])
Dtflite_convert.py86 output_arrays = _parse_array(flags.output_arrays)
91 "output_arrays": output_arrays
232 not flags.output_arrays):
Dconvert.py366 model.output_arrays.append(tensor_name(output_tensor))
373 def toco_convert_graph_def(input_data, input_arrays_with_shape, output_arrays, argument
416 for name in output_arrays:
417 model_flags.output_arrays.append(name)
Dconvert_saved_model.py223 output_arrays, tag_set, signature_key): argument
269 out_tensors = _get_tensors(graph, outputs, output_arrays)
Dconvert_test.py136 output_arrays = ["output"]
140 output_arrays,
180 output_arrays = ["output"]
185 output_arrays,
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_lib.py141 def evaluate_frozen_graph(filename, input_arrays, output_arrays): argument
163 sess.graph, output_arrays)
255 output_arrays, argument
281 filename, input_arrays, output_arrays, input_shapes)
290 output_arrays)
320 output_arrays, argument
341 filename, input_arrays, output_arrays, input_shapes)
344 tf_eval_func = evaluate_frozen_graph(filename, input_arrays, output_arrays)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lite.-t-f-lite-converter.pbtxt7 …\'input_tensors\', \'output_tensors\', \'input_arrays_with_shape\', \'output_arrays\'], varargs=No…
15 …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 …=[\'cls\', \'saved_model_dir\', \'input_arrays\', \'input_shapes\', \'output_arrays\', \'tag_set\'…
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 …=[\'cls\', \'saved_model_dir\', \'input_arrays\', \'input_shapes\', \'output_arrays\', \'tag_set\'…
/external/tensorflow/tensorflow/lite/python/testdata/
DBUILD16 "--output_arrays=output",
26 "--output_arrays=output",
41 "--output_arrays=output",
/external/tensorflow/tensorflow/lite/toco/
Dmodel_cmdline_flags.cc59 Flag("output_arrays", parsed_flags.output_arrays.bind(), in ParseModelFlagsFromCommandLineFlags()
60 parsed_flags.output_arrays.default_value(), in ParseModelFlagsFromCommandLineFlags()
209 parsed_model_flags.output_arrays.specified())) in ReadModelFlagsFromCommandLineFlags()
216 if (parsed_model_flags.output_arrays.specified()) { in ReadModelFlagsFromCommandLineFlags()
217 std::vector<string> output_arrays = in ReadModelFlagsFromCommandLineFlags() local
218 absl::StrSplit(parsed_model_flags.output_arrays.value(), ','); in ReadModelFlagsFromCommandLineFlags()
219 for (const string& output_array : output_arrays) { in ReadModelFlagsFromCommandLineFlags()
Dmodel_flags.proto141 repeated string output_arrays = 2; field
165 // and --output_arrays. This makes little sense, is only useful to
170 // --input_arrays and --output_arrays. By default (if false), only
Dtooling_util.cc109 for (const auto& output_array : model.flags.output_arrays()) { in IsOutputArray()
843 for (const string& output_array : model_flags.output_arrays()) { in CheckInputArraysAreNotOutputArrays()
886 for (const string& output_array : model_flags.output_arrays()) { in CheckNonAsciiIOArrays()
908 for (const string& output_array : model.flags.output_arrays()) { in CheckNonExistentIOArrays()
960 for (const string& output_array : model->flags.output_arrays()) { in FixNoMissingArray()
1077 for (const string& output_array : model.flags.output_arrays()) { in CheckOperatorOrdering()
1258 for (const string& output_array_name : model->flags.output_arrays()) { in FixEdgeArrays()
1582 model->flags.mutable_output_arrays()->CopyFrom(model_flags.output_arrays()); in ResolveModelFlags()
Dargs.h129 Arg<string> output_arrays; member
/external/tensorflow/tensorflow/lite/g3doc/convert/
Dcmdline_examples.md51 --output_arrays=MobilenetV1/Predictions/Reshape_1
70 `--output_arrays` are an aggregated, alphabetized list of the inputs and outputs
113 --output_arrays=MobilenetV1/Predictions/Reshape_1 \
138 --output_arrays=MobilenetV1/Predictions/Reshape_1 \
161 --output_arrays=InceptionV1/Logits/Predictions/Reshape_1
169 The flag `output_arrays` takes in a comma-separated list of output arrays as
180 …--output_arrays=InceptionV1/InceptionV1/Mixed_3b/Branch_1/Conv2d_0a_1x1/Relu,InceptionV1/Inception…
202 --output_arrays=InceptionV1/InceptionV1/Mixed_3b/concat_v2
215 When specifying intermediate arrays as `--input_arrays` and `--output_arrays`,
248 --output_arrays=MobilenetV1/Predictions/Reshape_1
[all …]
Dpython_api.md75 output_arrays = ["MobilenetV1/Predictions/Softmax"]
78 graph_def_file, input_arrays, output_arrays)
98 `output_arrays`, `tag_set` and `signature_key`. Details of each parameter are
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dquantize.cc554 if (model->flags.output_arrays(i) == in Run()
567 model->flags.output_arrays(i), in Run()
579 model->flags.output_arrays(i), in Run()
665 if (model->flags.output_arrays(i) == output) { in Run()
670 i, LogName(*dequantize_op), model->flags.output_arrays(i), in Run()
Dremove_final_dequantize_op.cc50 if (output == model->flags.output_arrays(i)) { in Run()
Dremove_unused_op.cc61 for (const string& output_array : model->flags.output_arrays()) { in Run()
Dremove_trivial_passthrough.cc147 for (const auto& array_name : model->flags.output_arrays()) { in RemoveTrivialPassthroughOp()
Dmove_binary_operator_before_reshape.cc144 for (const string& output_array : model->flags.output_arrays()) { in Run()
Dgraph_transformations.cc61 for (const string& output_array : model->flags.output_arrays()) { in DiscardUselessConnectedComponentsAndRNNBackEdges()
Ddequantize.cc136 for (const string& output_array : model->flags.output_arrays()) { in DequantizeArray()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc801 std::vector<llvm_ir::IrArray> output_arrays; in ConstructIrArrayForOutputs() local
804 output_arrays.reserve(num_outputs); in ConstructIrArrayForOutputs()
806 output_arrays.push_back(GetIrArray(hlo, hlo, {i})); in ConstructIrArrayForOutputs()
809 output_arrays.push_back(GetIrArray(hlo, hlo)); in ConstructIrArrayForOutputs()
811 return output_arrays; in ConstructIrArrayForOutputs()
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_from_protos_test.py58 model_flags.output_arrays.append(TensorName(out_tensor))

12