Home
last modified time | relevance | path

Searched defs:real_axis (Results 1 – 20 of 20) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/topi/
H A Dreduction.h66 std::vector<int> real_axis; in GetRealAxis() local
89 inline Array<IterVar> MakeReduceAxes(const std::vector<int>& real_axis, const Tensor& data) { in MakeReduceAxes()
99 inline Array<PrimExpr> MakeReduceTargetShape(const std::vector<int>& real_axis, const Tensor& data, in MakeReduceTargetShape()
185 auto real_axis = GetRealAxis(static_cast<int>(ndim), axis); in CommReduce() local
208 auto real_axis = GetRealAxis(static_cast<int>(ndim), axis); in CommReduceIdx() local
/dports/misc/tvm/incubator-tvm-0.6.1/topi/include/topi/
H A Dreduction.h65 std::vector<int> real_axis; in GetRealAxis() local
89 inline Array<IterVar> MakeReduceAxes(const std::vector<int>& real_axis, const Tensor& data) { in MakeReduceAxes()
100 inline Array<Expr> MakeReduceTargetShape(const std::vector<int>& real_axis, in MakeReduceTargetShape()
193 auto real_axis = GetRealAxis(static_cast<int>(ndim), axis); in CommReduce() local
219 auto real_axis = GetRealAxis(static_cast<int>(ndim), axis); in CommReduceIdx() local
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/include/topi/
H A Dreduction.h65 std::vector<int> real_axis; in GetRealAxis() local
89 inline Array<IterVar> MakeReduceAxes(const std::vector<int>& real_axis, const Tensor& data) { in MakeReduceAxes()
100 inline Array<Expr> MakeReduceTargetShape(const std::vector<int>& real_axis, in MakeReduceTargetShape()
193 auto real_axis = GetRealAxis(static_cast<int>(ndim), axis); in CommReduce() local
219 auto real_axis = GetRealAxis(static_cast<int>(ndim), axis); in CommReduceIdx() local
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/
H A Dslice_channel-inl.h82 int real_axis = axis_; in Forward() local
118 int real_axis = axis_; in Backward() local
198 int real_axis = param_.axis; in InferShape() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/
H A Dslice_channel-inl.h82 int real_axis = axis_; in Forward() local
118 int real_axis = axis_; in Backward() local
198 int real_axis = param_.axis; in InferShape() local
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/mkldnn/
H A Dmkldnn_log_softmax.cc100 const int real_axis, in GetLogSoftmaxFwd()
169 const int real_axis, in GetLogSoftmaxBwd()
H A Dmkldnn_softmax.cc102 const int real_axis, in GetSoftmaxFwd()
167 const int real_axis, in GetSoftmaxBwd()
H A Dmkldnn_batch_norm-inl.h156 const int real_axis = mxnet::op::batchnorm::GetRealAxis(shape, param.axis); in MKLDNNBatchNormForward() local
355 const int real_axis = mxnet::op::batchnorm::GetRealAxis(shape, param.axis); in MKLDNNBatchNormBackward() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/mkldnn/
H A Dmkldnn_log_softmax.cc100 const int real_axis, in GetLogSoftmaxFwd()
169 const int real_axis, in GetLogSoftmaxBwd()
H A Dmkldnn_softmax.cc102 const int real_axis, in GetSoftmaxFwd()
167 const int real_axis, in GetSoftmaxBwd()
H A Dmkldnn_batch_norm-inl.h156 const int real_axis = mxnet::op::batchnorm::GetRealAxis(shape, param.axis); in MKLDNNBatchNormForward() local
355 const int real_axis = mxnet::op::batchnorm::GetRealAxis(shape, param.axis); in MKLDNNBatchNormBackward() local
/dports/graphics/gimp-app/gimp-2.10.30/app/display/
H A Dgimptoolgyroscope.c778 GimpVector3 real_axis; in gimp_tool_gyroscope_rotate() local
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/tensor/
H A Dinit_op.h714 int real_axis = -1; in RangeLikeShape() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/tensor/
H A Dinit_op.h714 int real_axis = -1; in RangeLikeShape() local
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/numpy/
H A Dnp_matrix_op-inl.h848 int real_axis; in HSplitOpForward() local
870 int real_axis; in HSplitOpBackward() local
H A Dnp_matrix_op.cc1368 int real_axis; in HSplitOpShape() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/numpy/
H A Dnp_matrix_op-inl.h848 int real_axis; in HSplitOpForward() local
870 int real_axis; in HSplitOpBackward() local
H A Dnp_matrix_op.cc1368 int real_axis; in HSplitOpShape() local
/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/src/top/tensor/
H A Dtransform.cc716 int real_axis; in SqueezeShape() local
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/src/top/tensor/
H A Dtransform.cc716 int real_axis; in SqueezeShape() local