Home
last modified time | relevance | path

Searched refs:bias_shape (Results 1 – 25 of 59) sorted by relevance

123

/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/thirdparty/onnx/onnx/optimizer/passes/
H A Dfuse_add_bias_into_conv.h54 auto bias_shape = orig_bias->sizes(); in runTransform() local
72 for (int i = 0; i < static_cast<int64_t>(bias_shape.size()); ++i) { in runTransform()
73 if (bias_shape[i].is_int) { in runTransform()
74 num_el *= bias_shape[i].dim; in runTransform()
84 if (rank < static_cast<int64_t>(bias_shape.size())) { in runTransform()
93 if (bias_shape.size() > 1) { in runTransform()
95 std::vector<int64_t> axes(bias_shape.size() - 1); in runTransform()
121 } else if (rank > static_cast<int64_t>(bias_shape.size()) + 1) { in runTransform()
125 bias_shape[1 + bias_shape.size() - static_cast<unsigned>(rank)].dim == in runTransform()
127 ONNX_ASSERT(bias_shape.size() > 1); in runTransform()
[all …]
H A Dfuse_matmul_add_bias_into_gemm.h71 auto bias_shape = orig_bias->sizes(); in runTransform() local
72 auto bias_dim = static_cast<int64_t>(bias_shape.size()); in runTransform()
75 if (bias_dim == 1 && bias_shape[0].is_int) { in runTransform()
77 bias_M = bias_shape[0].dim; in runTransform()
78 } else if (bias_dim == 2 && bias_shape[0].is_int && bias_shape[1].is_int) { in runTransform()
79 bias_N = bias_shape[0].dim; in runTransform()
80 bias_M = bias_shape[1].dim; in runTransform()
/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/src/compiler/
H A Dpattern_util.h46 const TShape& oshape, const TShape& bias_shape) { in MatchBroadcast1DAxis() argument
47 dim_t axis_dim = bias_shape.ndim(); in MatchBroadcast1DAxis()
48 for (dim_t i = bias_shape.ndim(); i != 0; --i, --axis_dim) { in MatchBroadcast1DAxis()
49 if (bias_shape[i - 1] != 1) break; in MatchBroadcast1DAxis()
53 return {oshape.ndim() - bias_shape.ndim(), 0}; in MatchBroadcast1DAxis()
58 if (bias_shape[i] != 1) return {-1, -1}; in MatchBroadcast1DAxis()
61 oshape.ndim() - bias_shape.ndim() + axis_dim); in MatchBroadcast1DAxis()
62 if (oshape[axis] != bias_shape[axis_dim]) return {-1, -1}; in MatchBroadcast1DAxis()
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/src/compiler/
H A Dpattern_util.h46 const TShape& oshape, const TShape& bias_shape) { in MatchBroadcast1DAxis() argument
47 dim_t axis_dim = bias_shape.ndim(); in MatchBroadcast1DAxis()
48 for (dim_t i = bias_shape.ndim(); i != 0; --i, --axis_dim) { in MatchBroadcast1DAxis()
49 if (bias_shape[i - 1] != 1) break; in MatchBroadcast1DAxis()
53 return {oshape.ndim() - bias_shape.ndim(), 0}; in MatchBroadcast1DAxis()
58 if (bias_shape[i] != 1) return {-1, -1}; in MatchBroadcast1DAxis()
61 oshape.ndim() - bias_shape.ndim() + axis_dim); in MatchBroadcast1DAxis()
62 if (oshape[axis] != bias_shape[axis_dim]) return {-1, -1}; in MatchBroadcast1DAxis()
/dports/science/py-chainer/chainer-7.8.0/chainer/links/connection/
H A Dscale.py40 def __init__(self, axis=1, W_shape=None, bias_term=False, bias_shape=None): argument
52 if bias_shape is None:
56 self.bias = bias.Bias(axis, bias_shape)
H A Dlocal_convolution_2d.py85 bias_shape = (self.out_channels, oh, ow,)
88 self.b.initialize(bias_shape)
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/topi/python/
H A Dtest_topi_group_conv2d.py69 bias_shape = get_const_tuple(bias.shape)
76 b_np = np.random.uniform(size=bias_shape).astype(dtype)
83 b_np = np.random.uniform(size=bias_shape).astype(dtype)
187 bias_shape = get_const_tuple(bias.shape)
194 b_np = np.random.uniform(size=bias_shape).astype(dtype)
207 b_np = np.random.uniform(size=bias_shape).astype(dtype)
308 bias_shape = get_const_tuple(bias.shape)
315 b_np = np.random.uniform(size=bias_shape).astype(dtype)
322 b_np = np.random.uniform(size=bias_shape).astype(dtype)
H A Dtest_topi_conv2d_int8.py130 bias_shape = get_const_tuple(bias.shape)
137 b_np = np.random.uniform(size=bias_shape).astype(dtype)
142 b_np = np.random.uniform(size=bias_shape).astype(dtype)
233 bias_shape = get_const_tuple(bias.shape)
240 b_np = np.random.uniform(size=bias_shape).astype(dtype)
251 b_np = np.random.uniform(size=bias_shape).astype(dtype)
342 bias_shape = get_const_tuple(bias.shape)
349 b_np = np.random.uniform(size=bias_shape).astype(dtype)
354 b_np = np.random.uniform(size=bias_shape).astype(dtype)
H A Dtest_topi_conv2d_nhwc_tensorcore.py66 bias_shape = get_const_tuple(bias.shape)
73 b_np = np.random.uniform(size=bias_shape).astype(dtype)
77 b_np = np.random.uniform(size=bias_shape).astype(dtype)
H A Dtest_topi_conv3d_ndhwc_tensorcore.py68 bias_shape = get_const_tuple(bias.shape)
75 b_np = np.random.uniform(size=bias_shape).astype(dtype)
79 b_np = np.random.uniform(size=bias_shape).astype(dtype)
H A Dtest_topi_conv2d_nhwc_winograd.py74 bias_shape = get_const_tuple(bias.shape)
81 b_np = np.random.uniform(size=bias_shape).astype(dtype)
85 b_np = np.random.uniform(size=bias_shape).astype(dtype)
H A Dtest_topi_conv2d_winograd.py67 bias_shape = get_const_tuple(bias.shape)
74 b_np = np.random.uniform(size=bias_shape).astype(dtype)
78 b_np = np.random.uniform(size=bias_shape).astype(dtype)
/dports/misc/tvm/incubator-tvm-0.6.1/topi/tests/python/
H A Dtest_topi_group_conv2d.py44 bias_shape = get_const_tuple(bias.shape)
51 b_np = np.random.uniform(size=bias_shape).astype(dtype)
56 b_np = np.random.uniform(size=bias_shape).astype(dtype)
115 bias_shape = get_const_tuple(bias.shape)
122 b_np = np.random.uniform(size=bias_shape).astype(dtype)
132 b_np = np.random.uniform(size=bias_shape).astype(dtype)
H A Dtest_topi_conv2d_winograd.py41 bias_shape = get_const_tuple(bias.shape)
48 b_np = np.random.uniform(size=bias_shape).astype(dtype)
52 b_np = np.random.uniform(size=bias_shape).astype(dtype)
H A Dtest_topi_conv2d_int8.py45 bias_shape = get_const_tuple(bias.shape)
52 b_np = np.random.uniform(size=bias_shape).astype(dtype)
62 b_np = np.random.uniform(size=bias_shape).astype(dtype)
H A Dtest_topi_deformable_conv2d.py41 bias_shape = get_const_tuple(bias.shape)
49 b_np = np.random.uniform(size=bias_shape).astype(dtype)
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/tests/python/
H A Dtest_topi_group_conv2d.py44 bias_shape = get_const_tuple(bias.shape)
51 b_np = np.random.uniform(size=bias_shape).astype(dtype)
56 b_np = np.random.uniform(size=bias_shape).astype(dtype)
115 bias_shape = get_const_tuple(bias.shape)
122 b_np = np.random.uniform(size=bias_shape).astype(dtype)
132 b_np = np.random.uniform(size=bias_shape).astype(dtype)
H A Dtest_topi_conv2d_winograd.py41 bias_shape = get_const_tuple(bias.shape)
48 b_np = np.random.uniform(size=bias_shape).astype(dtype)
52 b_np = np.random.uniform(size=bias_shape).astype(dtype)
H A Dtest_topi_conv2d_int8.py45 bias_shape = get_const_tuple(bias.shape)
52 b_np = np.random.uniform(size=bias_shape).astype(dtype)
62 b_np = np.random.uniform(size=bias_shape).astype(dtype)
/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/src/top/nn/
H A Dconvolution.cc106 TShape bias_shape({param.channels}); in Conv2DInferShape() local
110 bias_shape = ConvertLayout(bias_shape, default_bias_layout, in Conv2DInferShape()
113 NNVM_ASSIGN_INPUT_SHAPE(attrs, *in_shape, Conv2DParam::kBias, bias_shape); in Conv2DInferShape()
206 TShape bias_shape({param.channels}); in WinogradConv2DInferShape() local
210 bias_shape = ConvertLayout(bias_shape, default_bias_layout, in WinogradConv2DInferShape()
213 NNVM_ASSIGN_INPUT_SHAPE(attrs, *in_shape, WinogradConv2DParam::kBias, bias_shape); in WinogradConv2DInferShape()
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/src/top/nn/
H A Dconvolution.cc106 TShape bias_shape({param.channels}); in Conv2DInferShape() local
110 bias_shape = ConvertLayout(bias_shape, default_bias_layout, in Conv2DInferShape()
113 NNVM_ASSIGN_INPUT_SHAPE(attrs, *in_shape, Conv2DParam::kBias, bias_shape); in Conv2DInferShape()
206 TShape bias_shape({param.channels}); in WinogradConv2DInferShape() local
210 bias_shape = ConvertLayout(bias_shape, default_bias_layout, in WinogradConv2DInferShape()
213 NNVM_ASSIGN_INPUT_SHAPE(attrs, *in_shape, WinogradConv2DParam::kBias, bias_shape); in WinogradConv2DInferShape()
/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/integration/
H A Dtest_winograd_nnpack.py41 bias_shape = get_const_tuple(bias.shape)
48 b_np = np.random.uniform(size=bias_shape).astype(dtype)
52 b_np = np.random.uniform(size=bias_shape).astype(dtype)
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/integration/
H A Dtest_winograd_nnpack.py41 bias_shape = get_const_tuple(bias.shape)
48 b_np = np.random.uniform(size=bias_shape).astype(dtype)
52 b_np = np.random.uniform(size=bias_shape).astype(dtype)
/dports/www/chromium-legacy/chromium-88.0.4324.182/chrome/browser/resource_coordinator/tab_ranker/
H A Dnative_inference.cc34 const int32_t* __restrict bias_shape, in FullyConnected() argument
41 CHECK_EQ(bias_shape[0], num_outputs); in FullyConnected()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/integration/
H A Dtest_winograd_nnpack.py57 bias_shape = get_const_tuple(bias.shape)
64 b_np = np.random.uniform(size=bias_shape).astype(dtype)
68 b_np = np.random.uniform(size=bias_shape).astype(dtype)

123