/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/numpy/linalg/ |
H A D | np_norm.cc | 55 SHAPE_ASSIGN_CHECK(*out_attrs, 0, TShape(0, -1)); in NumpyLpNormShape() 57 SHAPE_ASSIGN_CHECK(*out_attrs, 0, in NumpyLpNormShape() 87 SHAPE_ASSIGN_CHECK(*out_attrs, 0, out_shape); in NumpyMatrixNormShape() 118 std::vector<int>* out_attrs) { in NumpyNormType() argument 120 CHECK_EQ(out_attrs->size(), 4U); in NumpyNormType() 133 TYPE_ASSIGN_CHECK(*out_attrs, i, out_type); in NumpyNormType() 135 return out_attrs->at(0) != -1; in NumpyNormType() 140 mxnet::ShapeVector *out_attrs) { in NumpyNormShape() argument 149 assign_svd_empty(out_attrs); in NumpyNormShape() 161 assign_svd_empty(out_attrs); in NumpyNormShape() [all …]
|
H A D | np_eig.cc | 33 mxnet::ShapeVector *out_attrs) { in EigOpShape() argument 35 CHECK_EQ(out_attrs->size(), 2U); in EigOpShape() 37 const mxnet::TShape& eig_shape = (*out_attrs)[0]; in EigOpShape() 38 const mxnet::TShape& eigv_shape = (*out_attrs)[1]; in EigOpShape() 54 SHAPE_ASSIGN_CHECK(*out_attrs, 0, eig_shape); in EigOpShape() 56 SHAPE_ASSIGN_CHECK(*out_attrs, 1, a_shape); in EigOpShape() 84 SHAPE_ASSIGN_CHECK(*out_attrs, 1, eigv_shape); in EigOpShape() 97 SHAPE_ASSIGN_CHECK(*out_attrs, 0, eig_shape); in EigOpShape() 105 std::vector<int>* out_attrs) { in EigOpType() argument 107 CHECK_EQ(out_attrs->size(), 2U); in EigOpType() [all …]
|
H A D | np_pinv.cc | 32 mxnet::ShapeVector *out_attrs) { in PinvOpShape() argument 34 CHECK_EQ(out_attrs->size(), 1U); in PinvOpShape() 37 const mxnet::TShape& pinv_shape = (*out_attrs)[0]; in PinvOpShape() 77 std::vector<int>* out_attrs) { in PinvOpType() argument 79 CHECK_EQ(out_attrs->size(), 1U); in PinvOpType() 88 TYPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)); in PinvOpType() 92 return out_attrs->at(0) != -1; in PinvOpType() 120 CHECK_EQ(out_attrs->size(), 1U); in PinvScalarRcondOpShape() 122 const mxnet::TShape& pinv_shape = (*out_attrs)[0]; in PinvScalarRcondOpShape() 160 CHECK_EQ(out_attrs->size(), 1U); in PinvScalarRcondOpType() [all …]
|
H A D | np_tensorsolve.cc | 31 mxnet::ShapeVector *out_attrs) { in TensorsolveOpShape() argument 33 CHECK_EQ(out_attrs->size(), 1U); in TensorsolveOpShape() 46 SHAPE_ASSIGN_CHECK(*out_attrs, 0, b_shape); in TensorsolveOpShape() 51 SHAPE_ASSIGN_CHECK(*out_attrs, 0, a_shape); in TensorsolveOpShape() 56 SHAPE_ASSIGN_CHECK(*out_attrs, 0, a_shape); in TensorsolveOpShape() 82 SHAPE_ASSIGN_CHECK(*out_attrs, 0, out_shape); in TensorsolveOpShape() 90 std::vector<int>* out_attrs) { in TensorsolveOpType() argument 92 CHECK_EQ(out_attrs->size(), 1U); in TensorsolveOpType() 101 TYPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(1)); in TensorsolveOpType() 103 TYPE_ASSIGN_CHECK(*out_attrs, 0, mshadow::kFloat64); in TensorsolveOpType() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/numpy/linalg/ |
H A D | np_norm.cc | 55 SHAPE_ASSIGN_CHECK(*out_attrs, 0, TShape(0, -1)); in NumpyLpNormShape() 57 SHAPE_ASSIGN_CHECK(*out_attrs, 0, in NumpyLpNormShape() 87 SHAPE_ASSIGN_CHECK(*out_attrs, 0, out_shape); in NumpyMatrixNormShape() 118 std::vector<int>* out_attrs) { in NumpyNormType() argument 120 CHECK_EQ(out_attrs->size(), 4U); in NumpyNormType() 133 TYPE_ASSIGN_CHECK(*out_attrs, i, out_type); in NumpyNormType() 135 return out_attrs->at(0) != -1; in NumpyNormType() 140 mxnet::ShapeVector *out_attrs) { in NumpyNormShape() argument 149 assign_svd_empty(out_attrs); in NumpyNormShape() 161 assign_svd_empty(out_attrs); in NumpyNormShape() [all …]
|
H A D | np_eig.cc | 33 mxnet::ShapeVector *out_attrs) { in EigOpShape() argument 35 CHECK_EQ(out_attrs->size(), 2U); in EigOpShape() 37 const mxnet::TShape& eig_shape = (*out_attrs)[0]; in EigOpShape() 38 const mxnet::TShape& eigv_shape = (*out_attrs)[1]; in EigOpShape() 54 SHAPE_ASSIGN_CHECK(*out_attrs, 0, eig_shape); in EigOpShape() 56 SHAPE_ASSIGN_CHECK(*out_attrs, 1, a_shape); in EigOpShape() 84 SHAPE_ASSIGN_CHECK(*out_attrs, 1, eigv_shape); in EigOpShape() 97 SHAPE_ASSIGN_CHECK(*out_attrs, 0, eig_shape); in EigOpShape() 105 std::vector<int>* out_attrs) { in EigOpType() argument 107 CHECK_EQ(out_attrs->size(), 2U); in EigOpType() [all …]
|
H A D | np_pinv.cc | 32 mxnet::ShapeVector *out_attrs) { in PinvOpShape() argument 34 CHECK_EQ(out_attrs->size(), 1U); in PinvOpShape() 37 const mxnet::TShape& pinv_shape = (*out_attrs)[0]; in PinvOpShape() 77 std::vector<int>* out_attrs) { in PinvOpType() argument 79 CHECK_EQ(out_attrs->size(), 1U); in PinvOpType() 88 TYPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)); in PinvOpType() 92 return out_attrs->at(0) != -1; in PinvOpType() 120 CHECK_EQ(out_attrs->size(), 1U); in PinvScalarRcondOpShape() 122 const mxnet::TShape& pinv_shape = (*out_attrs)[0]; in PinvScalarRcondOpShape() 160 CHECK_EQ(out_attrs->size(), 1U); in PinvScalarRcondOpType() [all …]
|
H A D | np_tensorsolve.cc | 31 mxnet::ShapeVector *out_attrs) { in TensorsolveOpShape() argument 33 CHECK_EQ(out_attrs->size(), 1U); in TensorsolveOpShape() 46 SHAPE_ASSIGN_CHECK(*out_attrs, 0, b_shape); in TensorsolveOpShape() 51 SHAPE_ASSIGN_CHECK(*out_attrs, 0, a_shape); in TensorsolveOpShape() 56 SHAPE_ASSIGN_CHECK(*out_attrs, 0, a_shape); in TensorsolveOpShape() 82 SHAPE_ASSIGN_CHECK(*out_attrs, 0, out_shape); in TensorsolveOpShape() 90 std::vector<int>* out_attrs) { in TensorsolveOpType() argument 92 CHECK_EQ(out_attrs->size(), 1U); in TensorsolveOpType() 101 TYPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(1)); in TensorsolveOpType() 103 TYPE_ASSIGN_CHECK(*out_attrs, 0, mshadow::kFloat64); in TensorsolveOpType() [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/numpy/ |
H A D | np_cumsum.cc | 32 mxnet::ShapeVector *out_attrs) { in CumsumShape() argument 34 CHECK_EQ(out_attrs->size(), 1U); in CumsumShape() 41 SHAPE_ASSIGN_CHECK(*out_attrs, 0, out_shape); in CumsumShape() 42 return shape_is_known(out_attrs->at(0)); in CumsumShape() 48 std::vector<int> *out_attrs) { in CumsumType() argument 50 CHECK_EQ(out_attrs->size(), 1U); in CumsumType() 56 TYPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)); in CumsumType() 57 TYPE_ASSIGN_CHECK(*in_attrs, 0, out_attrs->at(0)); in CumsumType() 58 if (out_attrs->at(0) == mshadow::kBool) { in CumsumType() 59 (*out_attrs)[0] = mshadow::kInt64; in CumsumType() [all …]
|
H A D | np_moments_op.cc | 41 CHECK_EQ(out_attrs->size(), 2U); in NumpyMomentsShape() 47 SHAPE_ASSIGN_CHECK(*out_attrs, 0, out_shape); in NumpyMomentsShape() 48 SHAPE_ASSIGN_CHECK(*out_attrs, 1, out_shape); in NumpyMomentsShape() 50 return shape_is_known(out_attrs->at(0)) && shape_is_known(out_attrs->at(1)); in NumpyMomentsShape() 57 CHECK_EQ(out_attrs->size(), 2U); in NumpyMomentsType() 66 TYPE_ASSIGN_CHECK(*out_attrs, 1, in_attrs->at(0)); in NumpyMomentsType() 132 CHECK_EQ(out_attrs->size(), 2U); in NumpyWeightedAverageType() 134 TYPE_ASSIGN_CHECK(*in_attrs, 0, out_attrs->at(0)); in NumpyWeightedAverageType() 135 TYPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)); in NumpyWeightedAverageType() 139 TYPE_ASSIGN_CHECK(*out_attrs, 1, in_attrs->at(0)); in NumpyWeightedAverageType() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/numpy/ |
H A D | np_cumsum.cc | 32 mxnet::ShapeVector *out_attrs) { in CumsumShape() argument 34 CHECK_EQ(out_attrs->size(), 1U); in CumsumShape() 41 SHAPE_ASSIGN_CHECK(*out_attrs, 0, out_shape); in CumsumShape() 42 return shape_is_known(out_attrs->at(0)); in CumsumShape() 48 std::vector<int> *out_attrs) { in CumsumType() argument 50 CHECK_EQ(out_attrs->size(), 1U); in CumsumType() 56 TYPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)); in CumsumType() 57 TYPE_ASSIGN_CHECK(*in_attrs, 0, out_attrs->at(0)); in CumsumType() 58 if (out_attrs->at(0) == mshadow::kBool) { in CumsumType() 59 (*out_attrs)[0] = mshadow::kInt64; in CumsumType() [all …]
|
H A D | np_moments_op.cc | 41 CHECK_EQ(out_attrs->size(), 2U); in NumpyMomentsShape() 47 SHAPE_ASSIGN_CHECK(*out_attrs, 0, out_shape); in NumpyMomentsShape() 48 SHAPE_ASSIGN_CHECK(*out_attrs, 1, out_shape); in NumpyMomentsShape() 50 return shape_is_known(out_attrs->at(0)) && shape_is_known(out_attrs->at(1)); in NumpyMomentsShape() 57 CHECK_EQ(out_attrs->size(), 2U); in NumpyMomentsType() 66 TYPE_ASSIGN_CHECK(*out_attrs, 1, in_attrs->at(0)); in NumpyMomentsType() 132 CHECK_EQ(out_attrs->size(), 2U); in NumpyWeightedAverageType() 134 TYPE_ASSIGN_CHECK(*in_attrs, 0, out_attrs->at(0)); in NumpyWeightedAverageType() 135 TYPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)); in NumpyWeightedAverageType() 139 TYPE_ASSIGN_CHECK(*out_attrs, 1, in_attrs->at(0)); in NumpyWeightedAverageType() [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | elemwise_op_common.h | 72 dispatched = storage_type_assign(out_attrs, kCSRStorage, in ElemwiseStorageAttr() 86 dispatch_fallback(out_attrs, dispatch_mode); in ElemwiseStorageAttr() 109 CHECK_EQ(out_attrs->size(), n_out); in ElemwiseStorageType() 124 size_t out_size = out_attrs->size(); in ElemwiseAttrHelper() 131 CHECK_LE(out_size, out_attrs->size()); in ElemwiseAttrHelper() 142 deduce(*out_attrs, out_size, "output"); in ElemwiseAttrHelper() 153 write(out_attrs, out_size, "output"); in ElemwiseAttrHelper() 186 attrs, in_attrs, out_attrs, mxnet::TShape()); in ElemwiseShape() 192 std::vector<int> *out_attrs) { in ElemwiseType() argument 200 attrs, in_attrs, out_attrs, -1); in ElemwiseType() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | elemwise_op_common.h | 72 dispatched = storage_type_assign(out_attrs, kCSRStorage, in ElemwiseStorageAttr() 86 dispatch_fallback(out_attrs, dispatch_mode); in ElemwiseStorageAttr() 109 CHECK_EQ(out_attrs->size(), n_out); in ElemwiseStorageType() 124 size_t out_size = out_attrs->size(); in ElemwiseAttrHelper() 131 CHECK_LE(out_size, out_attrs->size()); in ElemwiseAttrHelper() 142 deduce(*out_attrs, out_size, "output"); in ElemwiseAttrHelper() 153 write(out_attrs, out_size, "output"); in ElemwiseAttrHelper() 186 attrs, in_attrs, out_attrs, mxnet::TShape()); in ElemwiseShape() 192 std::vector<int> *out_attrs) { in ElemwiseType() argument 200 attrs, in_attrs, out_attrs, -1); in ElemwiseType() [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
H A D | histogram-inl.h | 77 mxnet::ShapeVector* out_attrs) { in HistogramOpShape() argument 83 CHECK_EQ(out_attrs->size(), 2U); in HistogramOpShape() 90 SHAPE_ASSIGN_CHECK(*out_attrs, 0, mxnet::TShape(1, bin_cnt)); in HistogramOpShape() 101 SHAPE_ASSIGN_CHECK(*out_attrs, 1, in_attrs->at(1)); in HistogramOpShape() 104 return shape_is_known(out_attrs->at(0)) && shape_is_known(out_attrs->at(1)) && in HistogramOpShape() 105 out_attrs->at(0).Size() == out_attrs->at(1).Size() - 1; in HistogramOpShape() 110 std::vector<int>* out_attrs) { in HistogramOpType() argument 111 CHECK_EQ(out_attrs->size(), 2U); in HistogramOpType() 113 TYPE_ASSIGN_CHECK(*out_attrs, 0, mshadow::kInt64); in HistogramOpType() 114 TYPE_ASSIGN_CHECK(*out_attrs, 1, in_attrs->at(0)); in HistogramOpType() [all …]
|
H A D | elemwise_binary_op.cc | 37 CHECK_EQ(out_attrs->size(), 1U) << " in operator " << attrs.name; in SparseSparseWithDenseResult() 40 auto& out_stype = out_attrs->at(0); in SparseSparseWithDenseResult() 55 dispatched = dispatch_fallback(out_attrs, dispatch_mode); in SparseSparseWithDenseResult() 64 std::vector<int> *out_attrs) { in BackwardUseInStorageType() argument 67 CHECK_EQ(out_attrs->size(), 2U); in BackwardUseInStorageType() 75 int& lhs_grad_stype = out_attrs->at(0); in BackwardUseInStorageType() 76 int& rhs_grad_stype = out_attrs->at(1); in BackwardUseInStorageType() 78 dispatched = storage_type_assign(out_attrs, kDefaultStorage, in BackwardUseInStorageType() 83 && common::ContainsOnlyStorage(*out_attrs, kRowSparseStorage)) { in BackwardUseInStorageType() 84 dispatched = storage_type_assign(out_attrs, kRowSparseStorage, in BackwardUseInStorageType() [all …]
|
H A D | amp_cast.h | 63 std::vector<int> *out_attrs) { in AMPCastType() argument 69 CHECK_EQ(out_attrs->size(), 1U); in AMPCastType() 71 TYPE_ASSIGN_CHECK(*out_attrs, 0, param.dtype); in AMPCastType() 73 TYPE_ASSIGN_CHECK(*out_attrs, 0, (*in_attrs)[0]); in AMPCastType() 80 std::vector<int> *out_attrs) { in AMPMultiCastType() argument 86 CHECK_EQ(out_attrs->size(), param.num_outputs); in AMPMultiCastType() 100 TYPE_ASSIGN_CHECK(*out_attrs, i, widest_type); in AMPMultiCastType() 102 TYPE_ASSIGN_CHECK(*out_attrs, i, (*in_attrs)[i]); in AMPMultiCastType() 114 CHECK_EQ(out_attrs->size(), param.num_outputs); in AMPMultiCastShape() 119 SHAPE_ASSIGN_CHECK(*out_attrs, i, (*in_attrs)[i]); in AMPMultiCastShape() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
H A D | histogram-inl.h | 77 mxnet::ShapeVector* out_attrs) { in HistogramOpShape() argument 83 CHECK_EQ(out_attrs->size(), 2U); in HistogramOpShape() 90 SHAPE_ASSIGN_CHECK(*out_attrs, 0, mxnet::TShape(1, bin_cnt)); in HistogramOpShape() 101 SHAPE_ASSIGN_CHECK(*out_attrs, 1, in_attrs->at(1)); in HistogramOpShape() 104 return shape_is_known(out_attrs->at(0)) && shape_is_known(out_attrs->at(1)) && in HistogramOpShape() 105 out_attrs->at(0).Size() == out_attrs->at(1).Size() - 1; in HistogramOpShape() 110 std::vector<int>* out_attrs) { in HistogramOpType() argument 111 CHECK_EQ(out_attrs->size(), 2U); in HistogramOpType() 113 TYPE_ASSIGN_CHECK(*out_attrs, 0, mshadow::kInt64); in HistogramOpType() 114 TYPE_ASSIGN_CHECK(*out_attrs, 1, in_attrs->at(0)); in HistogramOpType() [all …]
|
H A D | elemwise_binary_op.cc | 37 CHECK_EQ(out_attrs->size(), 1U) << " in operator " << attrs.name; in SparseSparseWithDenseResult() 40 auto& out_stype = out_attrs->at(0); in SparseSparseWithDenseResult() 55 dispatched = dispatch_fallback(out_attrs, dispatch_mode); in SparseSparseWithDenseResult() 64 std::vector<int> *out_attrs) { in BackwardUseInStorageType() argument 67 CHECK_EQ(out_attrs->size(), 2U); in BackwardUseInStorageType() 75 int& lhs_grad_stype = out_attrs->at(0); in BackwardUseInStorageType() 76 int& rhs_grad_stype = out_attrs->at(1); in BackwardUseInStorageType() 78 dispatched = storage_type_assign(out_attrs, kDefaultStorage, in BackwardUseInStorageType() 83 && common::ContainsOnlyStorage(*out_attrs, kRowSparseStorage)) { in BackwardUseInStorageType() 84 dispatched = storage_type_assign(out_attrs, kRowSparseStorage, in BackwardUseInStorageType() [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/numpy/random/ |
H A D | dist_common.h | 153 SHAPE_ASSIGN_CHECK(*out_attrs, 0, TShape(oshape_vec)); in TwoparamsDistOpShape() 165 SHAPE_ASSIGN_CHECK(*out_attrs, 0, out); in TwoparamsDistOpShape() 168 SHAPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)) in TwoparamsDistOpShape() 171 SHAPE_ASSIGN_CHECK(*out_attrs, 0, TShape(0, -1)) in TwoparamsDistOpShape() 174 if (out_attrs->size() == 2U) { in TwoparamsDistOpShape() 175 SHAPE_ASSIGN_CHECK(*out_attrs, 1, out_attrs->at(0)); in TwoparamsDistOpShape() 199 SHAPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)) in UnaryDistOpShape() 201 SHAPE_ASSIGN_CHECK(*out_attrs, 0, TShape(0, -1)) in UnaryDistOpShape() 204 return shape_is_known(out_attrs->at(0)); in UnaryDistOpShape() 245 if (out_attrs->size() == 2U) { in TwoparamsDistOpConcatShape() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/numpy/random/ |
H A D | dist_common.h | 153 SHAPE_ASSIGN_CHECK(*out_attrs, 0, TShape(oshape_vec)); in TwoparamsDistOpShape() 165 SHAPE_ASSIGN_CHECK(*out_attrs, 0, out); in TwoparamsDistOpShape() 168 SHAPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)) in TwoparamsDistOpShape() 171 SHAPE_ASSIGN_CHECK(*out_attrs, 0, TShape(0, -1)) in TwoparamsDistOpShape() 174 if (out_attrs->size() == 2U) { in TwoparamsDistOpShape() 175 SHAPE_ASSIGN_CHECK(*out_attrs, 1, out_attrs->at(0)); in TwoparamsDistOpShape() 199 SHAPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)) in UnaryDistOpShape() 201 SHAPE_ASSIGN_CHECK(*out_attrs, 0, TShape(0, -1)) in UnaryDistOpShape() 204 return shape_is_known(out_attrs->at(0)); in UnaryDistOpShape() 245 if (out_attrs->size() == 2U) { in TwoparamsDistOpConcatShape() [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
H A D | dgl_graph.cc | 983 out_attrs->at(i) = gshape; in DGLSubgraphShape() 989 out_attrs->at(i) = gshape; in DGLSubgraphShape() 1211 CHECK_EQ(out_attrs->size(), 1U); in EdgeIDShape() 1226 CHECK_EQ(out_attrs->size(), 1U); in EdgeIDType() 1230 return out_attrs->at(0) != -1; in EdgeIDType() 1239 CHECK_EQ(out_attrs->size(), 1U); in EdgeIDStorageType() 1373 CHECK_EQ(out_attrs->size(), 1U); in DGLAdjacencyShape() 1384 CHECK_EQ(out_attrs->size(), 1U); in DGLAdjacencyType() 1387 return out_attrs->at(0) != -1; in DGLAdjacencyType() 1396 CHECK_EQ(out_attrs->size(), 1U); in DGLAdjacencyStorageType() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
H A D | dgl_graph.cc | 983 out_attrs->at(i) = gshape; in DGLSubgraphShape() 989 out_attrs->at(i) = gshape; in DGLSubgraphShape() 1211 CHECK_EQ(out_attrs->size(), 1U); in EdgeIDShape() 1226 CHECK_EQ(out_attrs->size(), 1U); in EdgeIDType() 1230 return out_attrs->at(0) != -1; in EdgeIDType() 1239 CHECK_EQ(out_attrs->size(), 1U); in EdgeIDStorageType() 1373 CHECK_EQ(out_attrs->size(), 1U); in DGLAdjacencyShape() 1384 CHECK_EQ(out_attrs->size(), 1U); in DGLAdjacencyType() 1387 return out_attrs->at(0) != -1; in DGLAdjacencyType() 1396 CHECK_EQ(out_attrs->size(), 1U); in DGLAdjacencyStorageType() [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/intgemm/ |
H A D | prepare_data_op.cc | 41 mxnet::ShapeVector* out_attrs) { in PrepareDataOpShape() argument 44 CHECK_EQ(out_attrs->size(), 1U); in PrepareDataOpShape() 46 SHAPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)); in PrepareDataOpShape() 47 SHAPE_ASSIGN_CHECK(*in_attrs, 0, out_attrs->at(0)); in PrepareDataOpShape() 51 return shape_is_known(out_attrs->at(0)); in PrepareDataOpShape() 56 std::vector<int>* out_attrs) { in PrepareDataOpType() argument 58 CHECK_EQ(out_attrs->size(), 1U); in PrepareDataOpType() 63 TYPE_ASSIGN_CHECK(*out_attrs, 0, mshadow::kInt8); in PrepareDataOpType() 71 std::vector<int>* out_attrs) { in PrepareDataOpStorageType() argument 73 CHECK_EQ(out_attrs->size(), 1U); in PrepareDataOpStorageType() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/intgemm/ |
H A D | prepare_data_op.cc | 41 mxnet::ShapeVector* out_attrs) { in PrepareDataOpShape() argument 44 CHECK_EQ(out_attrs->size(), 1U); in PrepareDataOpShape() 46 SHAPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0)); in PrepareDataOpShape() 47 SHAPE_ASSIGN_CHECK(*in_attrs, 0, out_attrs->at(0)); in PrepareDataOpShape() 51 return shape_is_known(out_attrs->at(0)); in PrepareDataOpShape() 56 std::vector<int>* out_attrs) { in PrepareDataOpType() argument 58 CHECK_EQ(out_attrs->size(), 1U); in PrepareDataOpType() 63 TYPE_ASSIGN_CHECK(*out_attrs, 0, mshadow::kInt8); in PrepareDataOpType() 71 std::vector<int>* out_attrs) { in PrepareDataOpStorageType() argument 73 CHECK_EQ(out_attrs->size(), 1U); in PrepareDataOpStorageType() [all …]
|