Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dgraphdef_to_tfl_flatbuffer.cc71 std::vector<string> output_arrays(model_flags.output_arrays().begin(), in ConvertGraphDefToTFLiteFlatBuffer() local
72 model_flags.output_arrays().end()); in ConvertGraphDefToTFLiteFlatBuffer()
74 tensorflow::ParseOutputArrayInfo(output_arrays, &specs.outputs)); in ConvertGraphDefToTFLiteFlatBuffer()
Dsaved_model_to_tfl_flatbuffer.cc97 function_output_names.reserve(model_flags.output_arrays().size()); in HandleInputOutputArraysWithModule()
105 if (function_output_names_size != model_flags.output_arrays().size()) { in HandleInputOutputArraysWithModule()
108 ", expected: ", model_flags.output_arrays().size()); in HandleInputOutputArraysWithModule()
113 for (const auto& output_array : model_flags.output_arrays()) { in HandleInputOutputArraysWithModule()
169 !model_flags.output_arrays().empty()) { in ConvertSavedModelToTFLiteFlatBuffer()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lite.-t-f-lite-converter.pbtxt10 …\'input_tensors\', \'output_tensors\', \'input_arrays_with_shape\', \'output_arrays\', \'experimen…
18 …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 …=[\'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/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc49 const std::vector<std::string>& output_arrays, in GraphdefToMlirImport() argument
71 TF_RETURN_IF_ERROR(ParseOutputArrayInfo(output_arrays, &specs.outputs)); in GraphdefToMlirImport()
107 const std::vector<std::string>& output_arrays, in GraphdefToMlirTranslateFunction() argument
114 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToMlirTranslateFunction()
126 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToMlirTranslateFunction() argument
137 TF_RETURN_IF_ERROR(ParseNodeNames(output_arrays, output_array_vector)); in GraphdefToMlirTranslateFunction()
221 const std::vector<std::string>& output_arrays, in GraphdefToSplattedMlirTranslateFunction() argument
228 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToSplattedMlirTranslateFunction()
270 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToSplattedMlirTranslateFunction() argument
281 TF_RETURN_IF_ERROR(ParseNodeNames(output_arrays, output_array_vector)); in GraphdefToSplattedMlirTranslateFunction()
Dtf_mlir_translate.h45 const std::vector<std::string>& output_arrays,
60 absl::string_view input_shapes, absl::string_view output_arrays,
73 const std::vector<std::string>& output_arrays,
87 absl::string_view input_shapes, absl::string_view output_arrays,
Dtf_mlir_translate_registration.cc48 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToMlirTranslateFunction()
62 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToSplattedMlirTranslateFunction()
/external/tensorflow/tensorflow/lite/python/
Dlite.py1375 output_arrays=self._output_arrays,
1505 output_arrays=None, argument
1529 if input_arrays or output_arrays:
1561 if output_arrays:
1562 output_tensors = _get_tensors_from_tensor_names(sess.graph, output_arrays)
1633 output_arrays=None, argument
1664 if not input_arrays_with_shape or not output_arrays:
1669 self._output_arrays = output_arrays
1676 if output_tensors is not None and output_arrays is not None:
1793 output_arrays=None, argument
[all …]
Dconvert_saved_model_test.py86 output_arrays=None, argument
98 output_arrays=output_arrays,
134 self._convertSavedModel(saved_model_dir, output_arrays=["invalid-output"])
161 output_arrays=["add"])
Dconvert.py524 model.output_arrays.append(output_tensor.name)
526 model.output_arrays.append(util.get_tensor_name(output_tensor))
541 def toco_convert_graph_def(input_data, input_arrays_with_shape, output_arrays, argument
588 for name in output_arrays:
589 model_flags.output_arrays.append(name)
Dconvert_saved_model.py156 output_arrays, tag_set, signature_key): argument
203 out_tensors = _get_tensors(graph, outputs, output_arrays)
Dtflite_convert.py99 output_arrays = _parse_array(flags.output_arrays)
104 "output_arrays": output_arrays
305 not flags.output_arrays):
Dutil.py252 output_arrays, argument
275 for array in output_arrays:
285 for array in input_arrays + output_arrays:
296 output_arrays = [get_tensor_name(tensor) for tensor in output_tensors]
298 sess, graph_def, output_arrays + hinted_outputs_nodes)
Dconvert_test.py107 output_arrays = ["output"]
111 output_arrays,
153 output_arrays = ["output"]
158 output_arrays,
/external/tensorflow/tensorflow/lite/python/testdata/
DBUILD26 "--output_arrays=output",
36 "--output_arrays=output",
51 "--output_arrays=output",
61 "--output_arrays=output",
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_lib.py327 def evaluate_frozen_graph(filename, input_arrays, output_arrays): argument
356 outputs = _util.get_tensors_from_tensor_names(graph, output_arrays)
589 output_arrays, argument
615 filename, input_arrays, output_arrays, input_shapes)
624 output_arrays,
671 output_arrays, argument
702 filename, input_arrays, output_arrays, input_shapes)
705 tf_eval_func = evaluate_frozen_graph(filename, input_arrays, output_arrays)
880 output_arrays, argument
904 output_arrays,
/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()
212 parsed_model_flags.output_arrays.specified())) in ReadModelFlagsFromCommandLineFlags()
219 if (parsed_model_flags.output_arrays.specified()) { in ReadModelFlagsFromCommandLineFlags()
220 std::vector<std::string> output_arrays = in ReadModelFlagsFromCommandLineFlags() local
221 absl::StrSplit(parsed_model_flags.output_arrays.value(), ','); in ReadModelFlagsFromCommandLineFlags()
222 for (const std::string& output_array : output_arrays) { in ReadModelFlagsFromCommandLineFlags()
/external/tensorflow/tensorflow/lite/g3doc/r1/convert/
Dcmdline_examples.md79 --output_arrays=MobilenetV1/Predictions/Reshape_1
84 and require additional flags `--input_arrays` and `--output_arrays` as this
101 --output_arrays=MobilenetV1/Predictions/Reshape_1 \
129 --output_arrays=MobilenetV1/Predictions/Reshape_1 \
154 --output_arrays=MobilenetV1/Predictions/Reshape_1 \
166 --output_arrays=MobilenetV1/Predictions/Reshape_1 \
184 --output_arrays=InceptionV1/Logits/Predictions/Reshape_1
189 The flag `--output_arrays` takes in a comma-separated list of output arrays as
198 …--output_arrays=InceptionV1/InceptionV1/Mixed_3b/Branch_1/Conv2d_0a_1x1/Relu,InceptionV1/Inception…
218 --output_arrays=InceptionV1/InceptionV1/Mixed_3b/concat_v2
[all …]
/external/tensorflow/tensorflow/lite/testing/
Dtoco_convert.py33 output_arrays, argument
61 " --output_arrays=%s" % ",".join(output_arrays))
174 output_arrays=output_tensors,
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_to_tfl_flatbuffer.cc102 absl::string_view input_shapes, absl::string_view output_arrays, in LoadFromGraphdefOrMlirSource() argument
124 input_shapes, output_arrays, /*control_output_arrays=*/"", in LoadFromGraphdefOrMlirSource()
131 input_shapes, output_arrays, /*control_output_arrays=*/"", in LoadFromGraphdefOrMlirSource()
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc77 absl::string_view output_arrays, bool prune_unused_nodes, in LoadFromGraphdefOrMlirSource() argument
96 input_shapes, output_arrays, /*control_output_arrays=*/"", in LoadFromGraphdefOrMlirSource()
Dtf_to_tfjs_json.h43 absl::string_view output_arrays, bool prune_unused_nodes,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc672 std::vector<llvm_ir::IrArray> output_arrays; in ConstructIrArrayForOutputs() local
675 output_arrays.reserve(num_outputs); in ConstructIrArrayForOutputs()
677 output_arrays.push_back(GetIrArray(hlo, hlo, {i})); in ConstructIrArrayForOutputs()
680 output_arrays.push_back(GetIrArray(hlo, hlo)); in ConstructIrArrayForOutputs()
682 return output_arrays; in ConstructIrArrayForOutputs()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dquantize.cc600 if (model->flags.output_arrays(i) == in Run()
613 model->flags.output_arrays(i), in Run()
625 model->flags.output_arrays(i), in Run()
710 if (model->flags.output_arrays(i) == output) { in Run()
715 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()

123