Home
last modified time | relevance | path

Searched refs:const_vars (Results 1 – 25 of 64) sorted by relevance

123

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/engine/
H A Dthreaded_engine.cc213 std::vector<VarHandle> const& const_vars, in NewOperator() argument
222 ret->const_vars.resize(const_vars.size()); in NewOperator()
225 std::transform(const_vars.begin(), const_vars.end(), in NewOperator()
230 CheckDuplicate(const_vars, mutable_vars); in NewOperator()
238 auto use = const_vars; in CheckDuplicate()
272 deps.reserve(threaded_opr->const_vars.size() + in DeleteOperator()
275 threaded_opr->const_vars.begin(), in DeleteOperator()
276 threaded_opr->const_vars.end()); in DeleteOperator()
298 threaded_opr->const_vars.size() + in Push()
305 for (auto&& i : threaded_opr->const_vars) { in Push()
[all …]
H A Dthreaded_engine.h238 std::vector<ThreadedVar*> const_vars; member
286 std::vector<VarHandle> const& const_vars,
294 std::vector<VarHandle> const& const_vars,
301 std::vector<VarHandle> const& const_vars,
441 std::vector<VarHandle> const_vars; member
453 void CheckDuplicate(std::vector<VarHandle> const& const_vars,
473 for (auto&& i : threaded_opr->const_vars) { in OnStart()
507 std::vector<VarHandle> const& const_vars, in BulkAppend() argument
519 bulk_status.const_vars.insert( in BulkAppend()
520 bulk_status.const_vars.end(), const_vars.begin(), const_vars.end()); in BulkAppend()
[all …]
H A Dnaive_engine.cc56 std::vector<VarHandle> const_vars; member
102 std::vector<VarHandle> const& const_vars, in NewOperator() argument
109 opr->const_vars = const_vars; in NewOperator()
141 opr->const_vars, in Push()
154 std::vector<VarHandle> const& const_vars, in PushAsync() argument
175 opr.reset(NewOperator(exec_fun, const_vars, mutable_vars, in PushAsync()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/engine/
H A Dthreaded_engine.cc213 std::vector<VarHandle> const& const_vars, in NewOperator() argument
222 ret->const_vars.resize(const_vars.size()); in NewOperator()
225 std::transform(const_vars.begin(), const_vars.end(), in NewOperator()
230 CheckDuplicate(const_vars, mutable_vars); in NewOperator()
238 auto use = const_vars; in CheckDuplicate()
272 deps.reserve(threaded_opr->const_vars.size() + in DeleteOperator()
275 threaded_opr->const_vars.begin(), in DeleteOperator()
276 threaded_opr->const_vars.end()); in DeleteOperator()
298 threaded_opr->const_vars.size() + in Push()
305 for (auto&& i : threaded_opr->const_vars) { in Push()
[all …]
H A Dthreaded_engine.h238 std::vector<ThreadedVar*> const_vars; member
286 std::vector<VarHandle> const& const_vars,
294 std::vector<VarHandle> const& const_vars,
301 std::vector<VarHandle> const& const_vars,
441 std::vector<VarHandle> const_vars; member
453 void CheckDuplicate(std::vector<VarHandle> const& const_vars,
473 for (auto&& i : threaded_opr->const_vars) { in OnStart()
507 std::vector<VarHandle> const& const_vars, in BulkAppend() argument
519 bulk_status.const_vars.insert( in BulkAppend()
520 bulk_status.const_vars.end(), const_vars.begin(), const_vars.end()); in BulkAppend()
[all …]
H A Dnaive_engine.cc56 std::vector<VarHandle> const_vars; member
102 std::vector<VarHandle> const& const_vars, in NewOperator() argument
109 opr->const_vars = const_vars; in NewOperator()
141 opr->const_vars, in Push()
154 std::vector<VarHandle> const& const_vars, in PushAsync() argument
175 opr.reset(NewOperator(exec_fun, const_vars, mutable_vars, in PushAsync()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/nnvm/
H A Dtvm_bridge.cc62 std::vector<Engine::VarHandle>* const_vars, in Init() argument
83 const_vars->push_back(nd.var()); in Init()
159 std::vector<Engine::VarHandle> const_vars, mutate_vars; in WrapAsyncCall() local
160 func->Init(args, const_loc, &const_vars, &mutate_vars); in WrapAsyncCall()
162 engine->DeduplicateVarHandle(&const_vars, &mutate_vars); in WrapAsyncCall()
165 }, func->ctx(), const_vars, mutate_vars); in WrapAsyncCall()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/nnvm/
H A Dtvm_bridge.cc62 std::vector<Engine::VarHandle>* const_vars, in Init() argument
83 const_vars->push_back(nd.var()); in Init()
159 std::vector<Engine::VarHandle> const_vars, mutate_vars; in WrapAsyncCall() local
160 func->Init(args, const_loc, &const_vars, &mutate_vars); in WrapAsyncCall()
162 engine->DeduplicateVarHandle(&const_vars, &mutate_vars); in WrapAsyncCall()
165 }, func->ctx(), const_vars, mutate_vars); in WrapAsyncCall()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/onnx/
H A Donnx_module.cc35 const Array<String>& const_vars) in ONNXSourceModuleNode() argument
36 : code_(code), symbol_(symbol), const_vars_(const_vars) {} in ONNXSourceModuleNode()
69 const Array<String>& const_vars) { in ONNXSourceModuleNodeCreate() argument
71 symbol.operator std::string(), const_vars); in ONNXSourceModuleNodeCreate()
/dports/lang/rust/rustc-1.58.1-src/compiler/rustc_infer/src/infer/
H A Dfudge.rs129 let const_vars = const_vars_since_snapshot( in fudge_inference_if_ok() localVariable
140 const_vars, in fudge_inference_if_ok()
160 && fudger.const_vars.0.is_empty() in fudge_inference_if_ok()
175 const_vars: (Range<ConstVid<'tcx>>, Vec<ConstVariableOrigin>), field
235 if self.const_vars.0.contains(&vid) { in fold_const()
238 let idx = (vid.index - self.const_vars.0.start.index) as usize; in fold_const()
239 let origin = self.const_vars.1[idx]; in fold_const()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/
H A Dmetadata_module.cc148 std::vector<std::vector<std::string>> const_vars; in SaveToBinary() local
151 const_vars.push_back(it.second); in SaveToBinary()
158 stream->Write(const_vars[i]); in SaveToBinary()
191 std::vector<std::vector<std::string>> const_vars; in LoadFromBinary() local
195 const_vars.push_back(vars); in LoadFromBinary()
200 sym_vars[symbols[i]] = const_vars[i]; in LoadFromBinary()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/ndarray/
H A Dndarray.cc601 std::vector<Engine::VarHandle> const_vars; in Reorder2DefaultAsync() local
608 }, ctx(), const_vars, mutable_vars, in Reorder2DefaultAsync()
627 std::vector<Engine::VarHandle> const_vars; in MKLDNNDataReorderAsync() local
639 }, ctx(), const_vars, mutable_vars, in MKLDNNDataReorderAsync()
814 std::vector<Engine::VarHandle> const_vars; in TernaryOp() local
825 }, lhs.ctx(), const_vars, { ret.var() }, in TernaryOp()
836 }, lhs.ctx(), const_vars, { ret.var() }, in TernaryOp()
878 return const_vars; in BinaryOpPrepare()
902 lhs.ctx(), const_vars, {ret.var()}, in BinaryOpKernel()
1309 const_vars.reserve(source.size()); in ElementwiseSum()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/ndarray/
H A Dndarray.cc601 std::vector<Engine::VarHandle> const_vars; in Reorder2DefaultAsync() local
608 }, ctx(), const_vars, mutable_vars, in Reorder2DefaultAsync()
627 std::vector<Engine::VarHandle> const_vars; in MKLDNNDataReorderAsync() local
639 }, ctx(), const_vars, mutable_vars, in MKLDNNDataReorderAsync()
814 std::vector<Engine::VarHandle> const_vars; in TernaryOp() local
825 }, lhs.ctx(), const_vars, { ret.var() }, in TernaryOp()
836 }, lhs.ctx(), const_vars, { ret.var() }, in TernaryOp()
878 return const_vars; in BinaryOpPrepare()
902 lhs.ctx(), const_vars, {ret.var()}, in BinaryOpKernel()
1309 const_vars.reserve(source.size()); in ElementwiseSum()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/target/source/
H A Dsource_module.cc113 const Array<String>& const_vars) in CSourceModuleNode() argument
114 : code_(code), fmt_(fmt), symbol_(symbol), const_vars_(const_vars) {} in CSourceModuleNode()
150 const Array<String>& const_vars) { in CSourceModuleCreate() argument
152 symbol.operator std::string(), const_vars); in CSourceModuleCreate()
212 .set_body_typed([](String code, String fmt, String symbol, Array<String> const_vars) { in __anondd16c2f70302() argument
213 return CSourceModuleCreate(code, fmt, symbol, const_vars); in __anondd16c2f70302()
H A Dcodegen_source_base.h145 const Array<String>& const_vars = {});
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mxnet/
H A Dengine.h168 std::vector<VarHandle> const& const_vars,
204 std::vector<VarHandle> const& const_vars,
264 std::vector<VarHandle> const& const_vars,
272 }, exec_ctx, const_vars, mutable_vars, prop, priority, opr_name);
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/include/mxnet/
H A Dengine.h168 std::vector<VarHandle> const& const_vars,
204 std::vector<VarHandle> const& const_vars,
264 std::vector<VarHandle> const& const_vars,
272 }, exec_ctx, const_vars, mutable_vars, prop, priority, opr_name);
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/kvstore/
H A Dkvstore_nccl.h220 std::vector<Engine::VarHandle> const_vars; in Reduce() local
262 const_vars.push_back(src[i].var()); in Reduce()
312 const_vars, in Reduce()
324 std::vector<Engine::VarHandle> const_vars; in Broadcast() local
363 const_vars.push_back(src.var()); in Broadcast()
416 const_vars, in Broadcast()
H A Dcomm.h141 std::vector<Engine::VarHandle> const_vars(src.size() - 1); in Reduce()
160 const_vars[i-1] = reduce[i].var(); in Reduce()
167 }, Context::CPU(), const_vars, {reduce[0].var()}, in Reduce()
172 std::vector<Engine::VarHandle> const_vars(src.size()); in Reduce()
188 const_vars[i] = reduce[i].var(); in Reduce()
199 }, Context::CPU(), const_vars, {buf_merged.var(), rsc.var}, in Reduce()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/kvstore/
H A Dkvstore_nccl.h220 std::vector<Engine::VarHandle> const_vars; in Reduce() local
262 const_vars.push_back(src[i].var()); in Reduce()
312 const_vars, in Reduce()
324 std::vector<Engine::VarHandle> const_vars; in Broadcast() local
363 const_vars.push_back(src.var()); in Broadcast()
416 const_vars, in Broadcast()
H A Dcomm.h141 std::vector<Engine::VarHandle> const_vars(src.size() - 1); in Reduce()
160 const_vars[i-1] = reduce[i].var(); in Reduce()
167 }, Context::CPU(), const_vars, {reduce[0].var()}, in Reduce()
172 std::vector<Engine::VarHandle> const_vars(src.size()); in Reduce()
188 const_vars[i] = reduce[i].var(); in Reduce()
199 }, Context::CPU(), const_vars, {buf_merged.var(), rsc.var}, in Reduce()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/
H A Doperator_util.cc651 std::vector<Engine::VarHandle> const_vars; in RegisterUnaryImperative() local
653 const_vars.push_back(src.var()); in RegisterUnaryImperative()
691 }, src.ctx(), const_vars, write_vars, in RegisterUnaryImperative()
922 std::vector<Engine::VarHandle> const_vars; in RegisterBinaryImperative() local
923 if (lhs.var() != ret.var()) const_vars.push_back(lhs.var()); in RegisterBinaryImperative()
924 if (rhs.var() != ret.var()) const_vars.push_back(rhs.var()); in RegisterBinaryImperative()
965 }, lhs.ctx(), const_vars, write_vars, in RegisterBinaryImperative()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/
H A Doperator_util.cc651 std::vector<Engine::VarHandle> const_vars; in RegisterUnaryImperative() local
653 const_vars.push_back(src.var()); in RegisterUnaryImperative()
691 }, src.ctx(), const_vars, write_vars, in RegisterUnaryImperative()
922 std::vector<Engine::VarHandle> const_vars; in RegisterBinaryImperative() local
923 if (lhs.var() != ret.var()) const_vars.push_back(lhs.var()); in RegisterBinaryImperative()
924 if (rhs.var() != ret.var()) const_vars.push_back(rhs.var()); in RegisterBinaryImperative()
965 }, lhs.ctx(), const_vars, write_vars, in RegisterBinaryImperative()
/dports/math/z3/z3-z3-4.8.13/src/muz/rel/
H A Ddl_mk_similarity_compressor.cpp393 ptr_vector<var> const_vars; //variables at indexes of their corresponding constants in merge_class() local
413 mod_var = const_vars[inf.parent_index()]; in merge_class()
415 const_vars.push_back(mod_var); in merge_class()
/dports/math/py-z3-solver/z3-z3-4.8.10/src/muz/rel/
H A Ddl_mk_similarity_compressor.cpp393 ptr_vector<var> const_vars; //variables at indexes of their corresponding constants in merge_class() local
413 mod_var = const_vars[inf.parent_index()]; in merge_class()
415 const_vars.push_back(mod_var); in merge_class()

123