File tree Expand file tree Collapse file tree 3 files changed +4
-8
lines changed Expand file tree Collapse file tree 3 files changed +4
-8
lines changed Original file line number Diff line number Diff line change @@ -442,7 +442,6 @@ struct Conv2DTransposeAttrs : public tvm::AttrsNode<Conv2DTransposeAttrs> {
442
442
}
443
443
};
444
444
445
- // //
446
445
/* ! \brief Attributes used in dilate operator */
447
446
struct DilateAttrs : public tvm ::AttrsNode<DilateAttrs> {
448
447
Array<IndexExpr> strides;
@@ -452,7 +451,6 @@ struct DilateAttrs : public tvm::AttrsNode<DilateAttrs> {
452
451
.describe (" Dilation stride on each dimension, 1 means no dilation." );
453
452
}
454
453
};
455
- // //
456
454
457
455
/* ! \brief Attributes used in 1D transposed convolution operator */
458
456
struct Conv1DTransposeAttrs : public tvm ::AttrsNode<Conv1DTransposeAttrs> {
Original file line number Diff line number Diff line change @@ -1355,7 +1355,7 @@ def dilate(data, strides):
1355
1355
data : tvm.relay.Expr
1356
1356
n-D, can be any layout.
1357
1357
1358
- strides : <tuple of n <int>
1358
+ strides : <tuple of <int>
1359
1359
Dilation stride on each dimension, 1 means no dilation.
1360
1360
1361
1361
Returns
Original file line number Diff line number Diff line change @@ -961,14 +961,13 @@ Do log on the data - do not accept logits.
961
961
.add_type_rel(" CrossEntropy" , CrossEntropyRel);
962
962
963
963
964
- // ///
965
964
// relay.nn.dilate
966
965
TVM_REGISTER_NODE_TYPE (DilateAttrs);
967
966
968
967
bool DilateRel (const Array<Type>& types,
969
- int num_inputs,
970
- const Attrs& attrs,
971
- const TypeReporter& reporter) {
968
+ int num_inputs,
969
+ const Attrs& attrs,
970
+ const TypeReporter& reporter) {
972
971
CHECK_EQ (types.size (), 2 );
973
972
const auto * x = types[0 ].as <TensorTypeNode>();
974
973
const DilateAttrs* param = attrs.as <DilateAttrs>();
@@ -1009,7 +1008,6 @@ Dilate data with zeros.
1009
1008
.add_argument(" x" , " 1D Tensor" , " Data to dilate." )
1010
1009
.set_support_level(10 )
1011
1010
.add_type_rel(" Dilate" , DilateRel);
1012
- // ///
1013
1011
1014
1012
// Positional relay function to create cross_entropy_with_logits operator used by frontend FFI.
1015
1013
Expr MakeCrossEntropyWithLogits (Expr predictions, Expr targets) {
You can’t perform that action at this time.
0 commit comments