We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 94d83ad commit 4cddc9dCopy full SHA for 4cddc9d
backends/cadence/fusion_g3/operators/op_mean.cpp
@@ -59,7 +59,7 @@ int prepare_data(
59
return num_axis_dims;
60
}
61
62
-Tensor& mean_dim_out(
+Tensor& mean_out(
63
KernelRuntimeContext& ctx,
64
const Tensor& in,
65
optional<ArrayRef<int64_t>> dim_list,
@@ -199,4 +199,4 @@ Tensor& mean_dim_out(
199
} // namespace native
200
} // namespace G3
201
} // namespace impl
202
-} // namespace cadence
+} // namespace cadence
backends/cadence/fusion_g3/operators/targets.bzl
@@ -36,9 +36,14 @@ OPERATORS = [
36
"add",
37
"cat",
38
"dequantize",
39
+ "div",
40
+ "exp",
41
+ "mean",
42
"mul",
43
"native_layer_norm",
44
+ "permute_copy",
45
"quantize",
46
+ "slice_copy",
47
"softmax",
48
"sub",
49
"div",
0 commit comments