/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/thirdparty/onnx/onnx/optimizer/passes/ |
H A D | fuse_add_bias_into_conv.h | 54 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 D | fuse_matmul_add_bias_into_gemm.h | 71 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 D | pattern_util.h | 46 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 D | pattern_util.h | 46 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 D | scale.py | 40 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 D | local_convolution_2d.py | 85 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 D | test_topi_group_conv2d.py | 69 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 D | test_topi_conv2d_int8.py | 130 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 D | test_topi_conv2d_nhwc_tensorcore.py | 66 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 D | test_topi_conv3d_ndhwc_tensorcore.py | 68 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 D | test_topi_conv2d_nhwc_winograd.py | 74 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 D | test_topi_conv2d_winograd.py | 67 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 D | test_topi_group_conv2d.py | 44 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 D | test_topi_conv2d_winograd.py | 41 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 D | test_topi_conv2d_int8.py | 45 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 D | test_topi_deformable_conv2d.py | 41 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 D | test_topi_group_conv2d.py | 44 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 D | test_topi_conv2d_winograd.py | 41 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 D | test_topi_conv2d_int8.py | 45 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 D | convolution.cc | 106 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 D | convolution.cc | 106 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 D | test_winograd_nnpack.py | 41 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 D | test_winograd_nnpack.py | 41 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 D | native_inference.cc | 34 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 D | test_winograd_nnpack.py | 57 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)
|