Home
last modified time | relevance | path

Searched refs:mxnet (Results 1 – 25 of 4075) sorted by relevance

12345678910>>...163

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/
H A Doperator_tune.cc27 namespace mxnet { namespace
118 ::mxnet::op::OperatorTune<__typ$>::ScheduleTune<__op$>( \
119 ::mxnet::op::UnaryOpTune<__typ$>::TuneBlankOperatorEx<__op$>)
133 ::mxnet::op::OperatorTune<__typ$>::ScheduleTune<__op$>( \
134 ::mxnet::op::UnaryOpTune<__typ$>::TuneUnaryOperator<__op$>)
143 bool ::mxnet::op::mxnet_op::tuned_op<::mxnet::op::mxnet_op::backward_grad_tuned<__op$>, __typ$>::\
164 ::mxnet::op::OperatorTune<__typ$>::ScheduleTune<__op$>( \
165 ::mxnet::op::BinaryOpTune<__typ$>::TuneBinaryOperator<__op$>)
174 bool ::mxnet::op::mxnet_op::tuned_op< \
175 ::mxnet::op::mxnet_op::backward_grad_tuned<__op$>, __typ$>:: \
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/
H A Doperator_tune.cc27 namespace mxnet { namespace
118 ::mxnet::op::OperatorTune<__typ$>::ScheduleTune<__op$>( \
119 ::mxnet::op::UnaryOpTune<__typ$>::TuneBlankOperatorEx<__op$>)
133 ::mxnet::op::OperatorTune<__typ$>::ScheduleTune<__op$>( \
134 ::mxnet::op::UnaryOpTune<__typ$>::TuneUnaryOperator<__op$>)
143 bool ::mxnet::op::mxnet_op::tuned_op<::mxnet::op::mxnet_op::backward_grad_tuned<__op$>, __typ$>::\
164 ::mxnet::op::OperatorTune<__typ$>::ScheduleTune<__op$>( \
165 ::mxnet::op::BinaryOpTune<__typ$>::TuneBinaryOperator<__op$>)
174 bool ::mxnet::op::mxnet_op::tuned_op< \
175 ::mxnet::op::mxnet_op::backward_grad_tuned<__op$>, __typ$>:: \
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/docs/python_docs/python/api/mxnet/
H A Dindex.rst18 mxnet chapter
29 mxnet.base
36 mxnet.gluon
37 mxnet.image
39 mxnet.io
45 mxnet.log
47 mxnet.model
50 mxnet.name
59 mxnet.rtc
62 mxnet.torch
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/docs/python_docs/python/api/mxnet/
H A Dindex.rst18 mxnet chapter
29 mxnet.base
36 mxnet.gluon
37 mxnet.image
39 mxnet.io
45 mxnet.log
47 mxnet.model
50 mxnet.name
59 mxnet.rtc
62 mxnet.torch
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/docs/static_site/src/pages/get_started/
H A Ddownload.md39 …mxnet/1.7.0/apache-mxnet-src-1.7.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
40 …mxnet/1.6.0/apache-mxnet-src-1.6.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
41 …mxnet/1.5.1/apache-mxnet-src-1.5.1-incubating.tar.gz) | [Download](https://archive.apache.org/di…
42 …mxnet/1.5.0/apache-mxnet-src-1.5.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
43 …mxnet/1.4.1/apache-mxnet-src-1.4.1-incubating.tar.gz) | [Download](https://archive.apache.org/di…
44 …mxnet/1.4.0/apache-mxnet-src-1.4.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
45 …mxnet/1.3.1/apache-mxnet-src-1.3.1-incubating.tar.gz) | [Download](https://archive.apache.org/di…
46 …mxnet/1.3.0/apache-mxnet-src-1.3.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
47 …mxnet/1.2.1/apache-mxnet-src-1.2.1-incubating.tar.gz) | [Download](https://archive.apache.org/di…
50 …mxnet/1.0.0/apache-mxnet-src-1.0.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/docs/static_site/src/pages/get_started/
H A Ddownload.md39 …mxnet/1.7.0/apache-mxnet-src-1.7.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
40 …mxnet/1.6.0/apache-mxnet-src-1.6.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
41 …mxnet/1.5.1/apache-mxnet-src-1.5.1-incubating.tar.gz) | [Download](https://archive.apache.org/di…
42 …mxnet/1.5.0/apache-mxnet-src-1.5.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
43 …mxnet/1.4.1/apache-mxnet-src-1.4.1-incubating.tar.gz) | [Download](https://archive.apache.org/di…
44 …mxnet/1.4.0/apache-mxnet-src-1.4.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
45 …mxnet/1.3.1/apache-mxnet-src-1.3.1-incubating.tar.gz) | [Download](https://archive.apache.org/di…
46 …mxnet/1.3.0/apache-mxnet-src-1.3.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
47 …mxnet/1.2.1/apache-mxnet-src-1.2.1-incubating.tar.gz) | [Download](https://archive.apache.org/di…
50 …mxnet/1.0.0/apache-mxnet-src-1.0.0-incubating.tar.gz) | [Download](https://archive.apache.org/di…
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/
H A Dlib_api.cc38 mxnet::ext::MXerrorMsgs* mxnet::ext::MXerrorMsgs::get() { in get()
492 mxnet::ext::Graph* mxnet::ext::Graph::fromJson(mxnet::ext::JsonVal val) { in fromJson()
564 mxnet::ext::JsonVal mxnet::ext::Graph::toJson() const { in toJson()
792 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setBackward(mxnet::ext::fcomp_t fgrad, in setBackward()
800 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setParseAttrs(mxnet::ext::parseAttrs_t func) { in setParseAttrs()
805 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setInferType(mxnet::ext::inferType_t func) { in setInferType()
810 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setInferSType(mxnet::ext::inferSType_t func) { in setInferSType()
815 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setInferShape(mxnet::ext::inferShape_t func) { in setInferShape()
820 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setMutateInputs(mxnet::ext::mutateInputs_t func) { in setMutateInputs()
825 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setCreateOpState(mxnet::ext::createOpState_t func, in setCreateOpState()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/
H A Dlib_api.cc38 mxnet::ext::MXerrorMsgs* mxnet::ext::MXerrorMsgs::get() { in get()
492 mxnet::ext::Graph* mxnet::ext::Graph::fromJson(mxnet::ext::JsonVal val) { in fromJson()
564 mxnet::ext::JsonVal mxnet::ext::Graph::toJson() const { in toJson()
792 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setBackward(mxnet::ext::fcomp_t fgrad, in setBackward()
800 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setParseAttrs(mxnet::ext::parseAttrs_t func) { in setParseAttrs()
805 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setInferType(mxnet::ext::inferType_t func) { in setInferType()
810 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setInferSType(mxnet::ext::inferSType_t func) { in setInferSType()
815 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setInferShape(mxnet::ext::inferShape_t func) { in setInferShape()
820 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setMutateInputs(mxnet::ext::mutateInputs_t func) { in setMutateInputs()
825 mxnet::ext::CustomOp& mxnet::ext::CustomOp::setCreateOpState(mxnet::ext::createOpState_t func, in setCreateOpState()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/docs/python_docs/python/api/
H A Dindex.rst34 :title: mxnet.ndarray
40 :title: mxnet.gluon
77 :title: mxnet.metric
101 :title: mxnet.random
113 :title: mxnet.sym
119 :title: mxnet.module
144 :title: mxnet.image
150 :title: mxnet.io
199 :title: mxnet.rtc
211 :title: mxnet.util
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/docs/python_docs/python/api/
H A Dindex.rst34 :title: mxnet.ndarray
40 :title: mxnet.gluon
77 :title: mxnet.metric
101 :title: mxnet.random
113 :title: mxnet.sym
119 :title: mxnet.module
144 :title: mxnet.image
150 :title: mxnet.io
199 :title: mxnet.rtc
211 :title: mxnet.util
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/cpp/thread_safety/
H A Dthread_safety_test.cc190 mxnet::cpp::Context ctx = mxnet::cpp::Context::gpu(0); in run_inference()
200 mxnet::cpp::Context ctx = mxnet::cpp::Context::cpu(0); in run_inference()
207 mxnet::cpp::Shape data_shape = mxnet::cpp::Shape(1, 3, 224, 224); in run_inference()
208 mxnet::cpp::Shape softmax_shape = mxnet::cpp::Shape(1); in run_inference()
367 mxnet::cpp::Context ctx = mxnet::cpp::Context::gpu(0); in run_inference_unsupported()
377 mxnet::cpp::Context ctx = mxnet::cpp::Context::cpu(0); in run_inference_unsupported()
385 mxnet::cpp::Shape softmax_shape = mxnet::cpp::Shape(1); in run_inference_unsupported()
541 mxnet::cpp::Context ctx = mxnet::cpp::Context::gpu(0); in TEST()
554 mxnet::cpp::Context ctx = mxnet::cpp::Context::cpu(0); in TEST()
582 mxnet::cpp::Symbol data = mxnet::cpp::Symbol::Variable("data"); in TEST()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/cpp/thread_safety/
H A Dthread_safety_test.cc190 mxnet::cpp::Context ctx = mxnet::cpp::Context::gpu(0); in run_inference()
200 mxnet::cpp::Context ctx = mxnet::cpp::Context::cpu(0); in run_inference()
207 mxnet::cpp::Shape data_shape = mxnet::cpp::Shape(1, 3, 224, 224); in run_inference()
208 mxnet::cpp::Shape softmax_shape = mxnet::cpp::Shape(1); in run_inference()
367 mxnet::cpp::Context ctx = mxnet::cpp::Context::gpu(0); in run_inference_unsupported()
377 mxnet::cpp::Context ctx = mxnet::cpp::Context::cpu(0); in run_inference_unsupported()
385 mxnet::cpp::Shape softmax_shape = mxnet::cpp::Shape(1); in run_inference_unsupported()
541 mxnet::cpp::Context ctx = mxnet::cpp::Context::gpu(0); in TEST()
554 mxnet::cpp::Context ctx = mxnet::cpp::Context::cpu(0); in TEST()
582 mxnet::cpp::Symbol data = mxnet::cpp::Symbol::Variable("data"); in TEST()
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/cpp/engine/
H A Dthreaded_engine_test.cc98 using namespace mxnet; in EvaluateWorkloads()
266 mxnet::NDArray *pnd = new mxnet::NDArray(ctx); in TEST()
348 [i](mxnet::RunContext ctx, mxnet::Engine::CallbackOnComplete cb) { in TEST()
362 engine->DeleteVariable([](mxnet::RunContext) {}, mxnet::Context{}, var); in TEST()
370 [i](mxnet::RunContext ctx, mxnet::Engine::CallbackOnComplete cb) { in TEST()
383 engine->DeleteVariable([](mxnet::RunContext) {}, mxnet::Context{}, var); in TEST()
389 engine->DeleteVariable([](mxnet::RunContext) {}, mxnet::Context{}, var); in TEST()
396 [](mxnet::RunContext ctx, mxnet::Engine::CallbackOnComplete cb) { in TEST()
409 engine->DeleteVariable([](mxnet::RunContext) {}, mxnet::Context{}, var); in TEST()
416 [](mxnet::RunContext ctx, mxnet::Engine::CallbackOnComplete cb) { in TEST()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/cpp/engine/
H A Dthreaded_engine_test.cc98 using namespace mxnet; in EvaluateWorkloads()
266 mxnet::NDArray *pnd = new mxnet::NDArray(ctx); in TEST()
348 [i](mxnet::RunContext ctx, mxnet::Engine::CallbackOnComplete cb) { in TEST()
362 engine->DeleteVariable([](mxnet::RunContext) {}, mxnet::Context{}, var); in TEST()
370 [i](mxnet::RunContext ctx, mxnet::Engine::CallbackOnComplete cb) { in TEST()
383 engine->DeleteVariable([](mxnet::RunContext) {}, mxnet::Context{}, var); in TEST()
389 engine->DeleteVariable([](mxnet::RunContext) {}, mxnet::Context{}, var); in TEST()
396 [](mxnet::RunContext ctx, mxnet::Engine::CallbackOnComplete cb) { in TEST()
409 engine->DeleteVariable([](mxnet::RunContext) {}, mxnet::Context{}, var); in TEST()
416 [](mxnet::RunContext ctx, mxnet::Engine::CallbackOnComplete cb) { in TEST()
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/tvmop/
H A Dufunc.cc35 namespace mxnet { namespace
130 .set_attr<mxnet::FCompute>("FCompute<gpu>", mxnet::op::TVMBinaryCompute<func_vadd_gpu>)
132 .set_attr<mxnet::FCompute>("FCompute<cpu>", mxnet::op::TVMBinaryCompute<func_vadd_cpu>)
140 .set_attr<mxnet::FCompute>("FCompute<gpu>",
143 .set_attr<mxnet::FCompute>("FCompute<cpu>",
212 .set_attr<mxnet::FInferShape>("FInferShape", mxnet::op::ElemwiseShape<1, 1>)
215 .set_attr<mxnet::FCompute>("FCompute<gpu>", mxnet::op::TVMUnaryCompute<func_deg2rad_gpu>)
217 .set_attr<mxnet::FCompute>("FCompute<cpu>", mxnet::op::TVMUnaryCompute<func_deg2rad_cpu>)
242 .set_attr<mxnet::FInferShape>("FInferShape", mxnet::op::ElemwiseShape<1, 1>)
245 .set_attr<mxnet::FCompute>("FCompute<gpu>", mxnet::op::TVMUnaryCompute<func_rad2deg_gpu>)
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/tvmop/
H A Dufunc.cc35 namespace mxnet { namespace
130 .set_attr<mxnet::FCompute>("FCompute<gpu>", mxnet::op::TVMBinaryCompute<func_vadd_gpu>)
132 .set_attr<mxnet::FCompute>("FCompute<cpu>", mxnet::op::TVMBinaryCompute<func_vadd_cpu>)
140 .set_attr<mxnet::FCompute>("FCompute<gpu>",
143 .set_attr<mxnet::FCompute>("FCompute<cpu>",
212 .set_attr<mxnet::FInferShape>("FInferShape", mxnet::op::ElemwiseShape<1, 1>)
215 .set_attr<mxnet::FCompute>("FCompute<gpu>", mxnet::op::TVMUnaryCompute<func_deg2rad_gpu>)
217 .set_attr<mxnet::FCompute>("FCompute<cpu>", mxnet::op::TVMUnaryCompute<func_deg2rad_cpu>)
242 .set_attr<mxnet::FInferShape>("FInferShape", mxnet::op::ElemwiseShape<1, 1>)
245 .set_attr<mxnet::FCompute>("FCompute<gpu>", mxnet::op::TVMUnaryCompute<func_rad2deg_gpu>)
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/scala-package/core/src/main/scala/org/apache/mxnet/javaapi/
H A DNDArray.scala18 package org.apache.mxnet.javaapi
20 import org.apache.mxnet.javaapi.DType.DType
51 = org.apache.mxnet.NDArray.empty(shape, ctx, dtype)
79 = org.apache.mxnet.NDArray.ones(shape, ctx, dtype)
92 = org.apache.mxnet.NDArray.full(shape, value, ctx)
175 = org.apache.mxnet.NDArray.lesserEqual(lhs, rhs)
177 = org.apache.mxnet.NDArray.lesserEqual(lhs, rhs)
179 = org.apache.mxnet.NDArray.lesserEqual(lhs, rhs)
189 = org.apache.mxnet.NDArray.array(
201 = org.apache.mxnet.NDArray.array(
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/scala-package/core/src/main/scala/org/apache/mxnet/javaapi/
H A DNDArray.scala18 package org.apache.mxnet.javaapi
20 import org.apache.mxnet.javaapi.DType.DType
51 = org.apache.mxnet.NDArray.empty(shape, ctx, dtype)
79 = org.apache.mxnet.NDArray.ones(shape, ctx, dtype)
92 = org.apache.mxnet.NDArray.full(shape, value, ctx)
175 = org.apache.mxnet.NDArray.lesserEqual(lhs, rhs)
177 = org.apache.mxnet.NDArray.lesserEqual(lhs, rhs)
179 = org.apache.mxnet.NDArray.lesserEqual(lhs, rhs)
189 = org.apache.mxnet.NDArray.array(
201 = org.apache.mxnet.NDArray.array(
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/example/multi_threaded_inference/
H A Dmulti_threaded_inference.cc130 void prepare_input_data(const mxnet::cpp::Shape& shape, const mxnet::cpp::Context& ctx, in prepare_input_data()
167 mxnet::Context backend_ctx; in run_inference()
168 mxnet::cpp::Context ctx = mxnet::cpp::Context::cpu(0); in run_inference()
177 mxnet::Context backend_ctx = mxnet::Context::CPU(0); in run_inference()
178 mxnet::cpp::Context ctx = mxnet::cpp::Context::cpu(0); in run_inference()
185 mxnet::cpp::Shape data_shape = mxnet::cpp::Shape(1, 3, 224, 224); in run_inference()
186 mxnet::cpp::Shape softmax_shape = mxnet::cpp::Shape(1); in run_inference()
278 mxnet::cpp::NDArray::WaitAll(); in run_inference()
297 mxnet::cpp::NDArray::WaitAll(); in run_inference()
338 mxnet::cpp::Shape input_shape = mxnet::cpp::Shape(1, 3, 224, 224); in main()
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/multi_threaded_inference/
H A Dmulti_threaded_inference.cc130 void prepare_input_data(const mxnet::cpp::Shape& shape, const mxnet::cpp::Context& ctx, in prepare_input_data()
167 mxnet::Context backend_ctx; in run_inference()
168 mxnet::cpp::Context ctx = mxnet::cpp::Context::cpu(0); in run_inference()
177 mxnet::Context backend_ctx = mxnet::Context::CPU(0); in run_inference()
178 mxnet::cpp::Context ctx = mxnet::cpp::Context::cpu(0); in run_inference()
185 mxnet::cpp::Shape data_shape = mxnet::cpp::Shape(1, 3, 224, 224); in run_inference()
186 mxnet::cpp::Shape softmax_shape = mxnet::cpp::Shape(1); in run_inference()
278 mxnet::cpp::NDArray::WaitAll(); in run_inference()
297 mxnet::cpp::NDArray::WaitAll(); in run_inference()
338 mxnet::cpp::Shape input_shape = mxnet::cpp::Shape(1, 3, 224, 224); in main()
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/numpy/random/
H A Ddist_common.h39 namespace mxnet {
49 inline int FillShape(const mxnet::TShape &lshape, const mxnet::TShape &rshape, in FillShape()
50 const mxnet::TShape &oshape, mxnet::TShape *new_lshape, in FillShape()
51 mxnet::TShape *new_rshape, mxnet::TShape *new_oshape) { in FillShape()
53 *new_lshape = mxnet::TShape(odim, 1); in FillShape()
54 *new_rshape = mxnet::TShape(odim, 1); in FillShape()
55 *new_oshape = mxnet::TShape(odim, 1); in FillShape()
104 if (!mxnet::dim_size_is_known(l) || !mxnet::dim_size_is_known(r)) continue; in CheckBroadcastable()
119 mxnet::TShape &out = (*out_ptr); in InferBroadcastShape()
126 if (!mxnet::dim_size_is_known(l) || !mxnet::dim_size_is_known(r)) continue; in InferBroadcastShape()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/numpy/random/
H A Ddist_common.h39 namespace mxnet {
49 inline int FillShape(const mxnet::TShape &lshape, const mxnet::TShape &rshape, in FillShape()
50 const mxnet::TShape &oshape, mxnet::TShape *new_lshape, in FillShape()
51 mxnet::TShape *new_rshape, mxnet::TShape *new_oshape) { in FillShape()
53 *new_lshape = mxnet::TShape(odim, 1); in FillShape()
54 *new_rshape = mxnet::TShape(odim, 1); in FillShape()
55 *new_oshape = mxnet::TShape(odim, 1); in FillShape()
104 if (!mxnet::dim_size_is_known(l) || !mxnet::dim_size_is_known(r)) continue; in CheckBroadcastable()
119 mxnet::TShape &out = (*out_ptr); in InferBroadcastShape()
126 if (!mxnet::dim_size_is_known(l) || !mxnet::dim_size_is_known(r)) continue; in InferBroadcastShape()
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/kvstore/
H A Dgradient_compression.cc31 namespace mxnet { namespace
76 mxnet::kvstore::split(s, ',', std::back_inserter(elems)); in DecodeParams()
101 void GradientCompression::Quantize(const mxnet::NDArray &from, mxnet::NDArray *to, in Quantize()
111 mxnet::Engine::Get()->PushSync([from, to, residual, threshold](mxnet::RunContext ctx) { in Quantize()
115 mxnet::FnProperty::kNormal, priority, "QuantizeCPU"); in Quantize()
119 mxnet::Engine::Get()->PushSync([from, to, residual, threshold](mxnet::RunContext ctx) { in Quantize()
125 mxnet::FnProperty::kNormal, priority, "QuantizeGPU"); in Quantize()
138 void GradientCompression::Dequantize(const mxnet::NDArray &from, mxnet::NDArray *to, in Dequantize()
147 mxnet::Engine::Get()->PushSync([from, to, threshold](mxnet::RunContext ctx) { in Dequantize()
151 mxnet::FnProperty::kNormal, priority, "DequantizeCPU"); in Dequantize()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/kvstore/
H A Dgradient_compression.cc31 namespace mxnet { namespace
76 mxnet::kvstore::split(s, ',', std::back_inserter(elems)); in DecodeParams()
101 void GradientCompression::Quantize(const mxnet::NDArray &from, mxnet::NDArray *to, in Quantize()
111 mxnet::Engine::Get()->PushSync([from, to, residual, threshold](mxnet::RunContext ctx) { in Quantize()
115 mxnet::FnProperty::kNormal, priority, "QuantizeCPU"); in Quantize()
119 mxnet::Engine::Get()->PushSync([from, to, residual, threshold](mxnet::RunContext ctx) { in Quantize()
125 mxnet::FnProperty::kNormal, priority, "QuantizeGPU"); in Quantize()
138 void GradientCompression::Dequantize(const mxnet::NDArray &from, mxnet::NDArray *to, in Dequantize()
147 mxnet::Engine::Get()->PushSync([from, to, threshold](mxnet::RunContext ctx) { in Dequantize()
151 mxnet::FnProperty::kNormal, priority, "DequantizeCPU"); in Dequantize()
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/cpp/include/
H A Dtest_tune.h52 namespace mxnet {
70 using shape_vect = mxnet::ShapeVector;
133 ::mxnet::op::tune::TuningMode fastest_standard_mode = ::mxnet::op::tune::kAuto;
135 if (i->second != ::mxnet::op::tune::kAuto) {
153 ::mxnet::op::tune::TuningMode closest_to;
155 closest_to = ::mxnet::op::tune::kNeverOMP;
157 closest_to = ::mxnet::op::tune::kAlwaysOMP;
192 const mxnet::TShape &shape = shapes[x];
198 const mxnet::TShape &lhs_shape = shapes[0];
260 using namespace mxnet::op;
[all …]

12345678910>>...163