/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | make_loss.cc | 42 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | convolution_v1.cc | 54 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | batch_norm_v1.cc | 42 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | pooling_v1.cc | 59 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | cross_device_copy.cc | 51 bool InferShape(mxnet::ShapeVector *in_shape, in InferShape() function in mxnet::op::CrossDeviceCopyProp
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | make_loss.cc | 42 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | convolution_v1.cc | 54 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | batch_norm_v1.cc | 42 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | pooling_v1.cc | 59 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | cross_device_copy.cc | 51 bool InferShape(mxnet::ShapeVector *in_shape, in InferShape() function in mxnet::op::CrossDeviceCopyProp
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
H A D | deformable_convolution.cc | 36 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | modulated_deformable_convolution.cc | 34 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | sync_batch_norm.cc | 41 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | count_sketch.cc | 39 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
H A D | deformable_convolution.cc | 36 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | modulated_deformable_convolution.cc | 34 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | sync_batch_norm.cc | 41 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/plugin/caffe/ |
H A D | caffe_loss.cc | 61 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | caffe_op.cc | 61 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/plugin/caffe/ |
H A D | caffe_loss.cc | 61 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
H A D | caffe_op.cc | 61 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/fusion/ |
H A D | fused_op.cc | 61 bool FusedOp::InferShape(const nnvm::NodeAttrs &attrs, in InferShape() function in mxnet::FusedOp 67 subgraph_ = mxnet::exec::InferShape(std::move(subgraph_), in InferShape() 184 return op->InferShape(attrs, in_attrs, out_attrs); in FusedOpInferShape()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/fusion/ |
H A D | fused_op.cc | 61 bool FusedOp::InferShape(const nnvm::NodeAttrs &attrs, in InferShape() function in mxnet::FusedOp 67 subgraph_ = mxnet::exec::InferShape(std::move(subgraph_), in InferShape() 184 return op->InferShape(attrs, in_attrs, out_attrs); in FusedOpInferShape()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mxnet/ |
H A D | operator.h | 200 virtual bool InferShape(mxnet::ShapeVector *in_shape, 266 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/include/mxnet/ |
H A D | operator.h | 200 virtual bool InferShape(mxnet::ShapeVector *in_shape, 266 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|