Searched defs:squeeze_dim (Results 1 – 8 of 8) sorted by relevance
81 int64_t squeeze_dim) { in BuildSqueezedDimensions()95 int64_t squeeze_dim) { in BuildUnsqueezedDimensions()
869 def squeeze_dim(func, *args, **kwargs): function
1306 auto squeeze_dim = rewriter.getI64ArrayAttr({-2}); in matchAndRewrite() local
756 for (auto& squeeze_dim : squeeze_dims_attr) { in matchAndRewrite() local
1745 for (auto& squeeze_dim : squeeze_dims_attr) { in matchAndRewrite() local
1652 int64_t squeeze_dim = squeeze_dim_apint.getSExtValue(); in verify() local
6477 int64_t squeeze_dim = squeeze_dim_apint.getSExtValue(); in matchAndRewrite() local
7055 int squeeze_dim = 2; in TEST_F() local