Skip to content

Add NHWC support in Portable Mean #19284

@digantdesai

Description

@digantdesai

hmm, NHWC support might not be there for portable.mean

ET_KERNEL_CHECK(
ctx, tensors_have_same_dim_order(in, out), InvalidArgument, out);
ET_KERNEL_CHECK(ctx, tensor_is_default_dim_order(in), InvalidArgument, out);

Originally posted by @digantdesai in #19139

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

Status

Ready

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions