Home
last modified time | relevance | path

Searched refs:ParseNodeShapes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dmlir_roundtrip_flags.cc61 TF_RETURN_IF_ERROR(ParseNodeShapes(shapes, node_shapes)); in ParseInputArrayInfo()
127 Status ParseNodeShapes( in ParseNodeShapes() function
Dmlir_roundtrip_flags.h104 Status ParseNodeShapes(
Dtf_mlir_translate.cc138 TF_RETURN_IF_ERROR(ParseNodeShapes(input_shapes, input_shapes_vector)); in GraphdefToMlirTranslateFunction()
282 TF_RETURN_IF_ERROR(ParseNodeShapes(input_shapes, input_shapes_vector)); in GraphdefToSplattedMlirTranslateFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtf_xla_mlir_translate.cc122 TF_RETURN_IF_ERROR(ParseNodeShapes(input_shapes_str, input_shapes_vector)); in ParseArgumentShapes()
192 tensorflow::ParseNodeShapes(input_shapes_str, input_shapes_vector)); in ParseXlaArguments()