Searched refs:ParseOutputArrayInfo (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | mlir_roundtrip_flags.h | 109 Status ParseOutputArrayInfo(absl::string_view array_names,
112 Status ParseOutputArrayInfo(const std::vector<string>& output_names,
|
| H A D | mlir_roundtrip_flags.cc | 69 Status ParseOutputArrayInfo(absl::string_view array_names, in ParseOutputArrayInfo() function
75 Status ParseOutputArrayInfo(const std::vector<string>& output_names, in ParseOutputArrayInfo() function
|
| H A D | tf_mlir_translate.cc | 73 TF_RETURN_IF_ERROR(ParseOutputArrayInfo(output_arrays, &specs.outputs)); in GraphdefToMlirImport()
75 ParseOutputArrayInfo(control_output_arrays, &specs.control_outputs)); in GraphdefToMlirImport()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
| H A D | graphdef_to_tfl_flatbuffer.cc | 77 tensorflow::ParseOutputArrayInfo(output_arrays, &specs.outputs)); in ConvertGraphDefToTFLiteFlatBuffer()
83 TF_RETURN_IF_ERROR(tensorflow::ParseOutputArrayInfo(control_output_arrays, in ConvertGraphDefToTFLiteFlatBuffer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
| H A D | flatbuffer_translate.cc | 139 if (!tensorflow::ParseOutputArrayInfo(input_arrays_flag, &inputs).ok()) { in FlatBufferFileToMlirTrans()
144 if (!tensorflow::ParseOutputArrayInfo(output_arrays_flag, &outputs).ok()) { in FlatBufferFileToMlirTrans()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | mlir_tf2xla.cc | 94 return ParseOutputArrayInfo(array_names, &specs->outputs); in ConvertOutputInfo()
|