Home
last modified time | relevance | path

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

123

/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
73 TF_RETURN_IF_ERROR(ParseOutputArrayInfo(output_arrays, &specs.outputs)); in GraphdefToMlirImport()
109 const std::vector<std::string>& output_arrays, in GraphdefToMlirTranslateFunction() argument
116 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToMlirTranslateFunction()
128 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToMlirTranslateFunction() argument
140 TF_RETURN_IF_ERROR(ParseNodeNames(output_arrays, output_array_vector)); in GraphdefToMlirTranslateFunction()
247 const std::vector<std::string>& output_arrays, in GraphdefToSplattedMlirTranslateFunction() argument
254 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToSplattedMlirTranslateFunction()
297 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToSplattedMlirTranslateFunction() argument
309 TF_RETURN_IF_ERROR(ParseNodeNames(output_arrays, output_array_vector)); in GraphdefToSplattedMlirTranslateFunction()
Dtf_mlir_translate.h47 const std::vector<std::string>& output_arrays,
63 absl::string_view input_shapes, absl::string_view output_arrays,
78 const std::vector<std::string>& output_arrays,
95 absl::string_view input_shapes, absl::string_view output_arrays,
Dtf_mlir_translate_registration.cc50 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToMlirTranslateFunction()
64 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToSplattedMlirTranslateFunction()
Dtf_mlir_translate_cl.h33 extern llvm::cl::opt<std::string> output_arrays;
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dgraphdef_to_tfl_flatbuffer.cc74 std::vector<string> output_arrays(model_flags.output_arrays().begin(), in ConvertGraphDefToTFLiteFlatBuffer() local
75 model_flags.output_arrays().end()); in ConvertGraphDefToTFLiteFlatBuffer()
77 tensorflow::ParseOutputArrayInfo(output_arrays, &specs.outputs)); in ConvertGraphDefToTFLiteFlatBuffer()
Dsaved_model_to_tfl_flatbuffer.cc103 function_output_names.reserve(model_flags.output_arrays().size()); in HandleInputOutputArraysWithModule()
112 if (function_output_names_size != model_flags.output_arrays().size()) { in HandleInputOutputArraysWithModule()
115 ", expected: ", model_flags.output_arrays().size()); in HandleInputOutputArraysWithModule()
120 for (const auto& output_array : model_flags.output_arrays()) { in HandleInputOutputArraysWithModule()
178 !model_flags.output_arrays().empty()) { in ConvertSavedModelToTFLiteFlatBuffer()
/external/tensorflow/tensorflow/lite/python/testdata/
DBUILD27 "--output_arrays=output",
37 "--output_arrays=output",
52 "--output_arrays=output",
62 "--output_arrays=output",
/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/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/python/
Dconvert_saved_model_test.py82 output_arrays=None, argument
94 output_arrays=output_arrays,
130 self._convertSavedModel(saved_model_dir, output_arrays=["invalid-output"])
157 output_arrays=["add"])
Dlite.py2092 output_arrays=self._output_arrays,
2243 output_arrays=None, argument
2267 if input_arrays or output_arrays:
2299 if output_arrays:
2300 output_tensors = _get_tensors_from_tensor_names(sess.graph, output_arrays)
2382 output_arrays=None, argument
2415 self._output_arrays = output_arrays
2422 if output_tensors is not None and output_arrays is not None:
2563 output_arrays=None, argument
2588 input_arrays_with_shape, output_arrays,
[all …]
Dconvert.py655 output_arrays, control_output_arrays, argument
702 if output_arrays:
703 for name in output_arrays:
704 model_flags.output_arrays.append(name)
788 model_flags.output_arrays.append(output_tensor.name)
790 model_flags.output_arrays.append(util.get_tensor_name(output_tensor))
Dconvert_saved_model.py156 output_arrays, tag_set, signature_key): argument
203 out_tensors = _get_tensors(graph, outputs, output_arrays)
Dtflite_convert.py124 output_arrays = _parse_array(flags.output_arrays)
129 "output_arrays": output_arrays
327 not flags.output_arrays):
Dutil.py222 output_arrays, argument
245 for array in output_arrays:
255 for array in input_arrays + output_arrays:
266 output_arrays = [get_tensor_name(tensor) for tensor in output_tensors]
268 sess, graph_def, output_arrays + hinted_outputs_nodes)
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h530 absl::Span<const llvm_ir::IrArray> output_arrays,
596 absl::Span<const llvm_ir::IrArray> output_arrays,
622 const IrEmitterUnnested::ReductionOutputMap& output_arrays,
630 llvm::Type* index_ty, const ReductionOutputMap& output_arrays,
637 llvm::Type* index_ty, const ReductionOutputMap& output_arrays,
Dir_emitter.cc662 std::vector<llvm_ir::IrArray> output_arrays; in ConstructIrArrayForOutputs() local
665 output_arrays.reserve(num_outputs); in ConstructIrArrayForOutputs()
667 output_arrays.push_back(GetIrArray(hlo, hlo, {i})); in ConstructIrArrayForOutputs()
670 output_arrays.push_back(GetIrArray(hlo, hlo)); in ConstructIrArrayForOutputs()
672 return output_arrays; in ConstructIrArrayForOutputs()
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc78 absl::string_view output_arrays, bool prune_unused_nodes, in LoadFromGraphdefOrMlirSource() argument
98 input_shapes, output_arrays, /*control_output_arrays=*/"", in LoadFromGraphdefOrMlirSource()
Dtf_to_tfjs_json.h44 absl::string_view output_arrays, bool prune_unused_nodes,
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_to_tfl_flatbuffer.cc112 absl::string_view input_shapes, absl::string_view output_arrays, in LoadFromGraphdefOrMlirSource() argument
136 input_shapes, output_arrays, control_output_arrays, in LoadFromGraphdefOrMlirSource()
144 input_shapes, output_arrays, control_output_arrays, in LoadFromGraphdefOrMlirSource()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dquantize.cc603 if (model->flags.output_arrays(i) == in Run()
616 model->flags.output_arrays(i), in Run()
628 model->flags.output_arrays(i), in Run()
713 if (model->flags.output_arrays(i) == output) { in Run()
718 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 std::string& output_array : model->flags.output_arrays()) { in Run()

123