Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc691 static constexpr char kAttrOutputShapes[] = "_output_shapes"; in ConvertUnsupportedOperator() local
748 if (HasAttr(node, kAttrOutputShapes)) { in ConvertUnsupportedOperator()
749 const auto& output_shapes = GetListAttr(node, kAttrOutputShapes); in ConvertUnsupportedOperator()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc62 constexpr char kAttrOutputShapes[] = "_output_shapes"; variable
84 const auto* attr = node->GetAttr(kAttrOutputShapes); in VerifyShapeAttributeMatch()
91 const auto* attr = node->GetAttr(kAttrOutputShapes); in VerifyShapeAttributeMatch()