Searched refs:CheckShapesAndTypes (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/fusion/ |
| H A D | fused_op.h | 143 void CheckShapesAndTypes(const std::vector<TBlob> &inputs,
|
| H A D | fused_op.cu | 706 void FusedOp::CheckShapesAndTypes(const std::vector<TBlob> &inputs, in CheckShapesAndTypes() function in mxnet::FusedOp
776 CheckShapesAndTypes(inputs, outputs, &in_dtypes, &in_ndims, in Forward()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/fusion/ |
| H A D | fused_op.h | 143 void CheckShapesAndTypes(const std::vector<TBlob> &inputs,
|
| H A D | fused_op.cu | 706 void FusedOp::CheckShapesAndTypes(const std::vector<TBlob> &inputs, in CheckShapesAndTypes() function in mxnet::FusedOp
776 CheckShapesAndTypes(inputs, outputs, &in_dtypes, &in_ndims, in Forward()
|