Home
last modified time | relevance | path

Searched defs:out_tensor (Results 1 – 23 of 23) sorted by relevance

/dports/science/bagel/bagel-1.2.2/src/smith/
H A Dsubtask.h58 std::shared_ptr<Tensor_<DataType>> out_tensor() { return out_; } in out_tensor() function
81 std::shared_ptr<Tensor_<DataType>> out_tensor(const size_t& i) { return out_[i]; } in out_tensor() function
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/apps/wasm-standalone/wasm-graph/src/
H A Dlib.rs80 let out_tensor: Tensor = output.into(); in run() localVariable
/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/cuda/rcnn/
H A Dproposal.py213 def calculate_overlap(out_tensor, box_a_idx, box_b_idx): argument
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/cuda/rcnn/
H A Dproposal.py226 def calculate_overlap(out_tensor, box_a_idx, box_b_idx): argument
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/cuda/rcnn/
H A Dproposal.py213 def calculate_overlap(out_tensor, box_a_idx, box_b_idx): argument
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/cuda/
H A Dnms.py255 def calculate_overlap(out_tensor, box_a_idx, box_b_idx): argument
/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/vision/rcnn/
H A Dproposal.py233 def calculate_overlap(out_tensor, box_a_idx, box_b_idx): argument
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/vision/rcnn/
H A Dproposal.py254 def calculate_overlap(out_tensor, box_a_idx, box_b_idx): argument
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/vision/rcnn/
H A Dproposal.py233 def calculate_overlap(out_tensor, box_a_idx, box_b_idx): argument
/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/cuda/
H A Dnms.py456 def calculate_overlap(out_tensor, box_a_idx, box_b_idx): argument
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/cuda/
H A Dnms.py456 def calculate_overlap(out_tensor, box_a_idx, box_b_idx): argument
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/quantization/
H A Dquantized_conv.cu164 Tensor<gpu, 1, DstType> out_tensor = out_.FlatTo1D<gpu, DstType>(s); in Forward() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/quantization/
H A Dquantized_conv.cu164 Tensor<gpu, 1, DstType> out_tensor = out_.FlatTo1D<gpu, DstType>(s); in Forward() local
/dports/misc/tvm/incubator-tvm-0.6.1/src/relay/backend/
H A Dinterpreter.cc444 for (auto out_tensor : outputs) { in ComputeDynamicShape() local
534 auto out_tensor = TensorValueNode::make( in InvokePrimitiveOp() local
571 Value out_tensor; in InvokePrimitiveOp() local
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/relay/backend/
H A Dinterpreter.cc444 for (auto out_tensor : outputs) { in ComputeDynamicShape() local
534 auto out_tensor = TensorValueNode::make( in InvokePrimitiveOp() local
571 Value out_tensor; in InvokePrimitiveOp() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/vm/
H A Dvm.cc541 auto out_tensor = NDArray::Empty({ndim}, {kDLInt, 64, 1}, {kDLCPU, 0}); in RunLoop() local
578 auto out_tensor = tensor_arr.CreateView(shape, tensor_arr->dtype); in RunLoop() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/plugin/warpctc/
H A Dwarpctc-inl.h98 Tensor<xpu, 2, float> out_tensor = out.FlatTo2D<xpu, float>(s); in Forward() local
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/plugin/warpctc/
H A Dwarpctc-inl.h98 Tensor<xpu, 2, float> out_tensor = out.FlatTo2D<xpu, float>(s); in Forward() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/backend/
H A Dinterpreter.cc389 for (auto out_tensor : outputs) { in ComputeDynamicShape() local
504 ObjectRef out_tensor; in InvokePrimitiveOp() local
/dports/science/py-dipy/dipy-1.4.1/dipy/workflows/
H A Dreconst.py227 out_dir='', out_tensor='tensors.nii.gz', out_fa='fa.nii.gz', argument
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/mkldnn/
H A Dmkldnn_transformer.cc202 const auto out_tensor = outputs[0]; in Initialize() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/mkldnn/
H A Dmkldnn_transformer.cc202 const auto out_tensor = outputs[0]; in Initialize() local
/dports/graphics/opencv/opencv-4.5.3/modules/gapi/src/backends/onnx/
H A Dgonnxbackend.cpp821 auto &out_tensor = std::get<1>(iter); in Run() local