Searched defs:new_axis_pos (Results 1 – 1 of 1) sorted by relevance
69 std::vector<int64_t> new_axis_pos = TenToTwo(GetValue<int64_t>(primitive->GetAttr(kNewAxisMask))); in EllipsisInferShape() local136 std::vector<int64_t> new_axis_pos = TenToTwo(GetValue<int64_t>(primitive->GetAttr(kNewAxisMask))); in ComputeInferShape() local