| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_mlir_translate.cc | 50 const std::vector<std::string>& control_output_arrays, in GraphdefToMlirImport() argument
73 ParseOutputArrayInfo(control_output_arrays, &specs.control_outputs)); in GraphdefToMlirImport()
108 const std::vector<std::string>& control_output_arrays, in GraphdefToMlirTranslateFunction() argument
114 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToMlirTranslateFunction()
127 absl::string_view control_output_arrays, bool prune_unused_nodes, in GraphdefToMlirTranslateFunction() argument
140 ParseNodeNames(control_output_arrays, control_output_array_vector)); in GraphdefToMlirTranslateFunction()
238 const std::vector<std::string>& control_output_arrays, in GraphdefToSplattedMlirTranslateFunction() argument
244 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToSplattedMlirTranslateFunction()
287 absl::string_view control_output_arrays, bool prune_unused_nodes, in GraphdefToSplattedMlirTranslateFunction() argument
300 ParseNodeNames(control_output_arrays, control_output_array_vector)); in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate.h | 47 const std::vector<std::string>& control_output_arrays,
62 absl::string_view control_output_arrays, bool prune_unused_nodes,
75 const std::vector<std::string>& control_output_arrays,
89 absl::string_view control_output_arrays, bool prune_unused_nodes,
|
| D | tf_mlir_translate_registration.cc | 48 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToMlirTranslateFunction()
62 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 | 93 std::vector<string> control_output_arrays( in ConvertGraphDefToTFLiteFlatBuffer() local
94 model_flags.control_output_arrays().begin(), in ConvertGraphDefToTFLiteFlatBuffer()
95 model_flags.control_output_arrays().end()); in ConvertGraphDefToTFLiteFlatBuffer()
96 TF_RETURN_IF_ERROR(tensorflow::ParseOutputArrayInfo(control_output_arrays, in ConvertGraphDefToTFLiteFlatBuffer()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | tf_to_tfl_flatbuffer.cc | 107 absl::string_view control_output_arrays, llvm::SourceMgr* source_mgr, in LoadFromGraphdefOrMlirSource() argument
129 input_shapes, output_arrays, control_output_arrays, in LoadFromGraphdefOrMlirSource()
136 input_shapes, output_arrays, control_output_arrays, in LoadFromGraphdefOrMlirSource()
|
| D | tf_to_tfl_flatbuffer.h | 46 absl::string_view control_output_arrays, llvm::SourceMgr* source_mgr,
|
| D | tf_tfl_translate.cc | 196 output_arrays, control_output_arrays, &source_mgr, &context); in main()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert.py | 706 enable_mlir_converter, control_output_arrays, *args, argument
760 if control_output_arrays:
761 for name in control_output_arrays:
762 model_flags.control_output_arrays.append(name)
|
| D | convert_test.py | 75 control_output_arrays=None,
114 control_output_arrays=None,
162 control_output_arrays=None,
|
| D | lite.py | 1741 control_output_arrays=self._control_output_arrays,
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model_flags.proto | 151 repeated string control_output_arrays = 24; field
|