Home
last modified time | relevance | path

Searched refs:ParseOutputArrayInfo (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dmlir_roundtrip_flags.h82 Status ParseOutputArrayInfo(absl::string_view array_names,
85 Status ParseOutputArrayInfo(const std::vector<string>& output_names,
Dmlir_roundtrip_flags.cc37 Status ParseOutputArrayInfo(absl::string_view array_names, in ParseOutputArrayInfo() function
43 Status ParseOutputArrayInfo(const std::vector<string>& output_names, in ParseOutputArrayInfo() function
Dtf_mlir_translate.cc71 TF_RETURN_IF_ERROR(ParseOutputArrayInfo(output_arrays, &specs.outputs)); in GraphdefToMlirImport()
73 ParseOutputArrayInfo(control_output_arrays, &specs.control_outputs)); in GraphdefToMlirImport()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_translate.cc134 if (!tensorflow::ParseOutputArrayInfo(input_arrays_flag, &inputs).ok()) { in FlatBufferFileToMlirTrans()
139 if (!tensorflow::ParseOutputArrayInfo(output_arrays_flag, &outputs).ok()) { in FlatBufferFileToMlirTrans()
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dgraphdef_to_tfl_flatbuffer.cc74 tensorflow::ParseOutputArrayInfo(output_arrays, &specs.outputs)); in ConvertGraphDefToTFLiteFlatBuffer()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dmlir_tf2xla.cc94 return ParseOutputArrayInfo(array_names, &specs->outputs); in ConvertOutputInfo()