|
|
@ -23,7 +23,7 @@ class RowwiseAddGradOpTest(GradientChecker):
|
|
|
|
"X": np.random.uniform(0.1, 1, [10, 10]).astype("float32"),
|
|
|
|
"X": np.random.uniform(0.1, 1, [10, 10]).astype("float32"),
|
|
|
|
"b": np.random.uniform(0.1, 1, [10, 1]).astype("float32")
|
|
|
|
"b": np.random.uniform(0.1, 1, [10, 1]).astype("float32")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
self.check_grad(op, inputs, set("X", "b"), "Out")
|
|
|
|
self.check_grad(op, inputs, set(["X", "b"]), "Out")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#TODO(dzh): rowwise_grad check
|
|
|
|
#TODO(dzh): rowwise_grad check
|
|
|
|