Searched refs:shrink_axis_pos (Results 1 – 2 of 2) sorted by relevance
70 …std::vector<int64_t> shrink_axis_pos = TenToTwo(GetValue<int64_t>(primitive->GetAttr(kShrinkAxisMa… in EllipsisInferShape() local103 if (j < shrink_axis_pos.size() && shrink_axis_pos[j] == 1) { in EllipsisInferShape()137 …std::vector<int64_t> shrink_axis_pos = TenToTwo(GetValue<int64_t>(primitive->GetAttr(kShrinkAxisMa… in ComputeInferShape() local173 if (j < shrink_axis_pos.size() && shrink_axis_pos[j] == 1) { in ComputeInferShape()
3283 shrink_axis_pos = bin(self.shrink_axis_mask)[-1:1:-1]3304 if j < len(shrink_axis_pos) and shrink_axis_pos[j] == '1':3340 if j < len(shrink_axis_pos) and shrink_axis_pos[j] == '1':