Searched refs:new_axis_pos (Results 1 – 2 of 2) sorted by relevance
69 std::vector<int64_t> new_axis_pos = TenToTwo(GetValue<int64_t>(primitive->GetAttr(kNewAxisMask))); in EllipsisInferShape() local71 (void)CheckAndConvertUtils::CheckInteger("infer", SizeToLong(new_axis_pos.size()), kGreaterEqual, in EllipsisInferShape()76 if (new_axis_pos[n] == 1) { in EllipsisInferShape()98 if (j < new_axis_pos.size() && new_axis_pos[j] == 1) { in EllipsisInferShape()136 std::vector<int64_t> new_axis_pos = TenToTwo(GetValue<int64_t>(primitive->GetAttr(kNewAxisMask))); in ComputeInferShape() local168 if (j < new_axis_pos.size() && new_axis_pos[j] == 1) { in ComputeInferShape()
3282 new_axis_pos = bin(self.new_axis_mask)[-1:1:-1]3300 if j < len(new_axis_pos) and new_axis_pos[j] == '1':3324 … len(tuple(filter(lambda x: x == '1', new_axis_pos[j + 1:slice_len])))3336 if j < len(new_axis_pos) and new_axis_pos[j] == '1':