HomeSort by relevance Sort by last modified time
    Searched refs:output_arrays (Results 1 - 18 of 18) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
remove_final_dequantize_op.cc 47 if (output == model->flags.output_arrays(i)) {
remove_unused_op.cc 58 for (const string& output_array : model->flags.output_arrays()) {
reorder_activation_functions.cc 70 if (model->flags.output_arrays(i) == ac_op->outputs[0]) {
graph_transformations.cc 61 for (const string& output_array : model->flags.output_arrays()) {
dequantize.cc 136 for (const string& output_array : model->flags.output_arrays()) {
quantize.cc 460 // If any of the model's output_arrays was pointing to the
464 if (model->flags.output_arrays(i) == dequantize_op->outputs[0]) {
531 if (model->flags.output_arrays(i) == output) {
fuse_binary_into_preceding_affine.cc 233 for (const string& output_array : model->flags.output_arrays()) {
  /external/tensorflow/tensorflow/contrib/lite/toco/
model_cmdline_flags.cc 55 "Deprecated: use --output_arrays instead. Name of the output array, "
59 Flag("output_arrays", parsed_flags.output_arrays.bind(),
60 parsed_flags.output_arrays.default_value(),
141 "and --output_arrays. This makes little sense, is only useful to "
146 "--input_arrays and --output_arrays. By default (if false), only "
186 parsed_model_flags.output_arrays.specified()))
193 if (parsed_model_flags.output_arrays.specified()) {
194 std::vector<string> output_arrays = local
195 absl::StrSplit(parsed_model_flags.output_arrays.value(), ',')
    [all...]
tooling_util.cc 78 for (const string& model_output : model.flags.output_arrays()) {
647 for (const string& output_array : model_flags.output_arrays()) {
650 << " is listed in both --input_arrays and --output_arrays.";
690 for (const string& output_array : model_flags.output_arrays()) {
692 << "Non-ASCII-printable character found in --output_arrays: "
707 for (const string& output_array : model.flags.output_arrays()) {
747 for (const string& output_array : model->flags.output_arrays()) {
861 for (const string& output_array : model.flags.output_arrays()) {
    [all...]
args.h 191 Arg<string> output_arrays; member in struct:toco::ParsedModelFlags
dump_graphviz.cc 98 for (const string& output_array : model.flags.output_arrays()) {
import_tensorflow.cc     [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/python/
toco_from_protos_test.py 57 model_flags.output_arrays.append(TensorName(out_tensor))
  /external/tensorflow/tensorflow/contrib/lite/python/
lite.py 208 model.output_arrays.append(_tensor_name(output_tensor))
  /external/tensorflow/tensorflow/contrib/lite/testing/
generate_examples.py 113 output_arrays,
121 output_arrays: name of the output tensors
139 " --output_arrays=%s" % ",".join(output_arrays))
148 output_arrays,
157 output_arrays: names of the output tensors
166 output_arrays=output_arrays,
314 output_arrays=output_tensors,
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/tflite/
export.cc 162 for (const string& output : model.flags.output_arrays()) {
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_unnested.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.cc     [all...]

Completed in 415 milliseconds