| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_mlir_translate.cc | 50 const std::vector<std::string>& control_output_arrays, in GraphdefToMlirImport() argument
75 ParseOutputArrayInfo(control_output_arrays, &specs.control_outputs)); in GraphdefToMlirImport()
110 const std::vector<std::string>& control_output_arrays, in GraphdefToMlirTranslateFunction() argument
116 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToMlirTranslateFunction()
129 absl::string_view control_output_arrays, bool prune_unused_nodes, in GraphdefToMlirTranslateFunction() argument
143 ParseNodeNames(control_output_arrays, control_output_array_vector)); in GraphdefToMlirTranslateFunction()
248 const std::vector<std::string>& control_output_arrays, in GraphdefToSplattedMlirTranslateFunction() argument
254 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToSplattedMlirTranslateFunction()
298 absl::string_view control_output_arrays, bool prune_unused_nodes, in GraphdefToSplattedMlirTranslateFunction() argument
312 ParseNodeNames(control_output_arrays, control_output_array_vector)); in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate.h | 48 const std::vector<std::string>& control_output_arrays,
64 absl::string_view control_output_arrays, bool prune_unused_nodes,
79 const std::vector<std::string>& control_output_arrays,
96 absl::string_view control_output_arrays, bool prune_unused_nodes,
|
| D | tf_mlir_translate_registration.cc | 50 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToMlirTranslateFunction()
64 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate_cl.h | 34 extern llvm::cl::opt<std::string> control_output_arrays;
|
| D | tf_mlir_translate_cl.cc | 52 opt<std::string> control_output_arrays( variable
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
| D | graphdef_to_tfl_flatbuffer.cc | 80 std::vector<string> control_output_arrays( in ConvertGraphDefToTFLiteFlatBuffer() local
81 model_flags.control_output_arrays().begin(), in ConvertGraphDefToTFLiteFlatBuffer()
82 model_flags.control_output_arrays().end()); in ConvertGraphDefToTFLiteFlatBuffer()
83 TF_RETURN_IF_ERROR(tensorflow::ParseOutputArrayInfo(control_output_arrays, in ConvertGraphDefToTFLiteFlatBuffer()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | tf_to_tfl_flatbuffer.cc | 113 absl::string_view control_output_arrays, llvm::SourceMgr* source_mgr, in LoadFromGraphdefOrMlirSource() argument
136 input_shapes, output_arrays, control_output_arrays, in LoadFromGraphdefOrMlirSource()
144 input_shapes, output_arrays, control_output_arrays, in LoadFromGraphdefOrMlirSource()
|
| D | tf_to_tfl_flatbuffer.h | 49 absl::string_view control_output_arrays, llvm::SourceMgr* source_mgr,
|
| D | tf_tfl_translate.cc | 227 output_arrays, control_output_arrays, &source_mgr, &context); in main()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert.py | 655 output_arrays, control_output_arrays, argument
705 if control_output_arrays:
706 for name in control_output_arrays:
707 model_flags.control_output_arrays.append(name)
|
| D | convert_test.py | 74 control_output_arrays=None,
112 control_output_arrays=None,
160 control_output_arrays=None,
|
| D | lite.py | 2093 control_output_arrays=self._control_output_arrays,
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model_flags.proto | 151 repeated string control_output_arrays = 24; field
|