Home
last modified time | relevance | path

Searched refs:node_shapes_str (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dmlir_roundtrip_flags.cc160 std::vector<string> node_shapes_str = absl::StrSplit(shapes_str, ':'); in ParseNodeShapes() local
161 for (int i = 0; i < node_shapes_str.size(); i++) { in ParseNodeShapes()
162 if (node_shapes_str[i] == "*") { in ParseNodeShapes()
168 absl::StrSplit(node_shapes_str[i], ',')) { in ParseNodeShapes()
/external/tensorflow/tensorflow/compiler/mlir/python/
Dmlir.cc168 std::vector<string> node_shapes_str = absl::StrSplit(input_data_shapes, ':'); in ImportGraphDef() local
172 for (const string &shape_str : node_shapes_str) { in ImportGraphDef()