Home
last modified time | relevance | path

Searched defs:ishape (Results 1 – 25 of 205) sorted by relevance

123456789

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/tensor/
H A Ddiag_op-inl.h63 inline mxnet::TShape DiagShapeImpl(const mxnet::TShape& ishape, const int k, in DiagShapeImpl()
115 const mxnet::TShape& ishape = (*in_attrs)[0]; in DiagOpShape() local
188 const mxnet::TShape& ishape, in DiagOpProcess()
298 const mxnet::TShape& ishape = inputs[0].shape_; in DiagOpForward() local
319 const mxnet::TShape& ishape = inputs[0].shape_; in DiagOpBackward() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/tensor/
H A Ddiag_op-inl.h63 inline mxnet::TShape DiagShapeImpl(const mxnet::TShape& ishape, const int k, in DiagShapeImpl()
115 const mxnet::TShape& ishape = (*in_attrs)[0]; in DiagOpShape() local
188 const mxnet::TShape& ishape, in DiagOpProcess()
298 const mxnet::TShape& ishape = inputs[0].shape_; in DiagOpForward() local
319 const mxnet::TShape& ishape = inputs[0].shape_; in DiagOpBackward() local
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/numpy/
H A Dnp_matrix_op-inl.h492 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyFlipForward() local
642 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyRot90ComputeFlipIml() local
709 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyRot90ComputeFlipTransposeIml() local
1002 const mxnet::TShape &ishape = (*in_attrs)[0]; in NumpyDiagOpShape() local
1069 const mxnet::TShape &ishape, in NumpyDiagOpImpl()
1129 const mxnet::TShape &ishape = inputs[0].shape_; in NumpyDiagOpForward() local
1151 const mxnet::TShape &ishape = inputs[0].shape_; in NumpyDiagOpBackward() local
1219 const mxnet::TShape& ishape = (*in_attrs)[0]; in NumpyDiagonalOpShape() local
1347 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyDiagonalOpForward() local
1369 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyDiagonalOpBackward() local
[all …]
H A Dnp_trace_op-inl.h94 const mxnet::TShape& ishape, in NumpyTraceOpProcess()
221 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyTraceOpForward() local
244 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyTraceOpBackward() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/numpy/
H A Dnp_matrix_op-inl.h492 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyFlipForward() local
642 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyRot90ComputeFlipIml() local
709 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyRot90ComputeFlipTransposeIml() local
1002 const mxnet::TShape &ishape = (*in_attrs)[0]; in NumpyDiagOpShape() local
1069 const mxnet::TShape &ishape, in NumpyDiagOpImpl()
1129 const mxnet::TShape &ishape = inputs[0].shape_; in NumpyDiagOpForward() local
1151 const mxnet::TShape &ishape = inputs[0].shape_; in NumpyDiagOpBackward() local
1219 const mxnet::TShape& ishape = (*in_attrs)[0]; in NumpyDiagonalOpShape() local
1347 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyDiagonalOpForward() local
1369 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyDiagonalOpBackward() local
[all …]
H A Dnp_trace_op-inl.h94 const mxnet::TShape& ishape, in NumpyTraceOpProcess()
221 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyTraceOpForward() local
244 const mxnet::TShape& ishape = inputs[0].shape_; in NumpyTraceOpBackward() local
/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/tests/python/frontend/caffe2/
H A Dtest_graph.py22 def compare_graph(init, predict, nnvm_sym, ishape): argument
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/topi/nn/
H A Dflatten.h51 auto ishape = x->shape; variable
H A Dbnn.h54 auto ishape = data->shape; variable
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/tests/python/frontend/caffe2/
H A Dtest_graph.py22 def compare_graph(init, predict, nnvm_sym, ishape): argument
/dports/misc/tvm/incubator-tvm-0.6.1/topi/include/topi/nn/
H A Dflatten.h53 auto ishape = x->shape; variable
H A Dbnn.h53 auto ishape = data->shape; variable
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/include/topi/nn/
H A Dflatten.h53 auto ishape = x->shape; variable
H A Dbnn.h53 auto ishape = data->shape; variable
/dports/misc/tvm/incubator-tvm-0.6.1/topi/tests/python/
H A Dtest_topi_l2norm.py24 def verify_l2_normalize(ishape, eps, axis=None): argument
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/tests/python/
H A Dtest_topi_l2norm.py24 def verify_l2_normalize(ishape, eps, axis=None): argument
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/backend/contrib/dnnl/
H A Dcodegen.cc62 auto ishape = GetShape(call->args[0]->checked_type()); in Conv2d() local
85 auto ishape = GetShape(call->args[0]->checked_type()); in Dense() local
98 auto ishape = GetShape(call->args[0]->checked_type()); in Relu() local
111 auto ishape = GetShape(call->args[0]->checked_type()); in BatchNorm() local
126 auto ishape = GetShape(call->args[0]->checked_type()); in Add() local
/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/tests/python/compiler/
H A Dtest_top_level1.py389 def verify_concatenate(ishape, axis): argument
404 def verify_split(ishape, indices_or_sections, axis): argument
419 def verify_strided_slice(ishape, begin, end, strideinp=None): argument
512 def verify_lrn(ishape, size, axis, bias, alpha, beta): argument
528 def verify_l2_normalize(ishape, eps, axis): argument
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/tests/python/compiler/
H A Dtest_top_level1.py389 def verify_concatenate(ishape, axis): argument
404 def verify_split(ishape, indices_or_sections, axis): argument
419 def verify_strided_slice(ishape, begin, end, strideinp=None): argument
512 def verify_lrn(ishape, size, axis, bias, alpha, beta): argument
528 def verify_l2_normalize(ishape, eps, axis): argument
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tools/accnn/
H A Drank_selection.py27 def calc_complexity(ishape, node): argument
/dports/misc/mxnet/incubator-mxnet-1.9.0/tools/accnn/
H A Drank_selection.py27 def calc_complexity(ishape, node): argument
/dports/devel/rth/rth/jpgraph-1.8/src/utils/jpdocgen/
H A Ddbschema_ddda.php39 var $ig, $img, $iscale, $ishape; variable in Driver
/dports/devel/rth/rth/jpgraph-1.8/jpgraph-1.18/src/Examples/
H A Ddbschemaex1.php39 var $ig, $img, $iscale, $ishape; variable in Driver
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mshadow/mshadow/extension/
H A Dreshape.h49 Shape<dimsrc> ishape = ShapeCheck<dimsrc, SrcExp>::Check(src_); in ReshapeExp() local
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/3rdparty/mshadow/mshadow/extension/
H A Dreshape.h49 Shape<dimsrc> ishape = ShapeCheck<dimsrc, SrcExp>::Check(src_); in ReshapeExp() local

123456789