Searched refs:squeeze_dims_attr (Results 1 – 4 of 4) sorted by relevance
1769 const auto* squeeze_dims_attr = node.GetAttr(kAttrSqueezeDims); in IsDimsSupported() local1770 if (squeeze_dims_attr == nullptr) { in IsDimsSupported()1774 IsAlongAxis(*squeeze_dims_attr, indices({'H', 'W'}), kRank)) || in IsDimsSupported()1776 IsAlongAxis(*squeeze_dims_attr, indices({'N', 'H', 'W'}), kRank)); in IsDimsSupported()1781 const auto* squeeze_dims_attr = node->GetAttr(kAttrSqueezeDims); in UpdateSqueezeDims() local1782 if (squeeze_dims_attr == nullptr) { in UpdateSqueezeDims()1788 const int squeeze_dims_size = squeeze_dims_attr->list().i_size(); in UpdateSqueezeDims()1791 int dim = squeeze_dims_attr->list().i(i); in UpdateSqueezeDims()
2254 const auto* squeeze_dims_attr = squeeze_node->GetAttr("squeeze_dims"); in TEST_F() local2255 const auto& list = squeeze_dims_attr->list(); in TEST_F()2317 const auto* squeeze_dims_attr = squeeze_node->GetAttr("squeeze_dims"); in TEST_F() local2318 const auto& list = squeeze_dims_attr->list(); in TEST_F()2381 const auto* squeeze_dims_attr = squeeze_node->GetAttr("squeeze_dims"); in TEST_F() local2382 const auto& list = squeeze_dims_attr->list(); in TEST_F()
749 auto squeeze_dims_attr = tf_squeeze_op.squeeze_dimsAttr(); in matchAndRewrite() local751 for (auto& squeeze_dim : squeeze_dims_attr) { in matchAndRewrite()
1667 auto squeeze_dims_attr = tfl_squeeze_op.squeeze_dimsAttr(); in matchAndRewrite() local1669 for (auto& squeeze_dim : squeeze_dims_attr) { in matchAndRewrite()