Home
last modified time | relevance | path

Searched refs:out_attrs (Results 1 – 25 of 484) sorted by relevance

12345678910>>...20

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/numpy/linalg/
H A Dnp_norm.cc55 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 Dnp_eig.cc33 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 Dnp_pinv.cc32 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 Dnp_tensorsolve.cc31 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 Dnp_norm.cc55 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 Dnp_eig.cc33 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 Dnp_pinv.cc32 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 Dnp_tensorsolve.cc31 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 Dnp_cumsum.cc32 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 Dnp_moments_op.cc41 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 Dnp_cumsum.cc32 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 Dnp_moments_op.cc41 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 Delemwise_op_common.h72 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 Delemwise_op_common.h72 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 Dhistogram-inl.h77 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 Delemwise_binary_op.cc37 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 Damp_cast.h63 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 Dhistogram-inl.h77 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 Delemwise_binary_op.cc37 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 Ddist_common.h153 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 Ddist_common.h153 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 Ddgl_graph.cc983 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 Ddgl_graph.cc983 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 Dprepare_data_op.cc41 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 Dprepare_data_op.cc41 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 …]

12345678910>>...20