| /dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/ |
| H A D | reduce_support.hpp | 30 static func_t make(const layout_t &src_layout, const layout_t &dst_layout) { in IR_DECL_DERIVED_TYPE_ID()
31 return func_t(new reduce_t(src_layout, dst_layout)); in IR_DECL_DERIVED_TYPE_ID()
38 return (src_layout == other.src_layout) in is_equal()
43 return ir_utils::get_hash(src_layout, dst_layout); in get_hash()
48 oss << "reduce[" << src_layout << ", " << dst_layout << "]"; in str()
55 layout_t src_layout;
59 reduce_t(const layout_t &src_layout, const layout_t &dst_layout) in reduce_t() argument
60 : src_layout(src_layout), dst_layout(dst_layout) {} in reduce_t()
|
| H A D | reorder_support.hpp | 111 static func_t make(const layout_t &src_layout, const layout_t &dst_layout, in IR_DECL_DERIVED_TYPE_ID()
113 return func_t(new reorder_t(src_layout, dst_layout, grf_perm)); in IR_DECL_DERIVED_TYPE_ID()
120 return (src_layout == other.src_layout) in is_equal()
126 return ir_utils::get_hash(src_layout, dst_layout, grf_perm.get()); in get_hash()
131 oss << "reorder[" << src_layout << ", " << dst_layout << "]"; in str()
138 layout_t src_layout;
144 reorder_t(const layout_t &src_layout, const layout_t &dst_layout, in reorder_t() argument
146 : src_layout(src_layout), dst_layout(dst_layout), grf_perm(grf_perm) {} in reorder_t()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/nnvm/src/top/nn/ |
| H A D | nn_common.h | 62 inline TShape ConvertLayout(TShape src, const Layout& src_layout, const Layout& dst_layout) { in ConvertLayout() argument
63 if (src_layout == dst_layout) { in ConvertLayout()
65 } else if (!src_layout.defined()) { in ConvertLayout()
68 LOG(FATAL) << "cannot convert " << src_layout << " to undefined layout"; in ConvertLayout()
71 CHECK(src_layout.convertible(dst_layout)) << "cannot convert from " in ConvertLayout()
72 << src_layout << " to " << dst_layout; in ConvertLayout()
75 for (size_t i = 0; i < src_layout.ndim(); ++i) { in ConvertLayout()
76 Layout::LayoutDim src_dim = src_layout[i]; in ConvertLayout()
80 int src_minor_pos = src_layout.indexof(Layout::to_subdim(src_dim)); in ConvertLayout()
81 int src_factor = src_layout.subsizeof(src_dim); in ConvertLayout()
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/src/top/nn/ |
| H A D | nn_common.h | 62 inline TShape ConvertLayout(TShape src, const Layout& src_layout, const Layout& dst_layout) { in ConvertLayout() argument
63 if (src_layout == dst_layout) { in ConvertLayout()
65 } else if (!src_layout.defined()) { in ConvertLayout()
68 LOG(FATAL) << "cannot convert " << src_layout << " to undefined layout"; in ConvertLayout()
71 CHECK(src_layout.convertible(dst_layout)) << "cannot convert from " in ConvertLayout()
72 << src_layout << " to " << dst_layout; in ConvertLayout()
75 for (size_t i = 0; i < src_layout.ndim(); ++i) { in ConvertLayout()
76 Layout::LayoutDim src_dim = src_layout[i]; in ConvertLayout()
80 int src_minor_pos = src_layout.indexof(Layout::to_subdim(src_dim)); in ConvertLayout()
81 int src_factor = src_layout.subsizeof(src_dim); in ConvertLayout()
[all …]
|
| /dports/science/py-chainer/chainer-7.8.0/chainer/ |
| H A D | memory_layouts.py | 46 def _transpose_array(arr, src_layout, dst_layout): argument
47 trans = _get_layout_transpose_axes(arr.ndim, src_layout, dst_layout)
53 def _transpose_shape(shape, src_layout, dst_layout): argument
54 trans = _get_layout_transpose_axes(len(shape), src_layout, dst_layout)
60 def _get_layout_transpose_axes(ndim, src_layout, dst_layout, inverse=False): argument
63 if src_layout == dst_layout:
68 assert src_layout == CUDNN_CHANNEL_FIRST_X
73 assert src_layout == CUDNN_CHANNEL_FIRST_W
76 elif src_layout == CUDNN_CHANNEL_LAST_X:
81 elif src_layout == CUDNN_CHANNEL_LAST_W:
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/lang/ |
| H A D | data_layout.cc | 204 const Layout& src_layout, in GetStoreRule() argument
206 if (!src_layout.defined() || src_layout.name().empty() || in GetStoreRule()
215 for (size_t j = 0; j < src_layout.ndim(); ++j) { in GetStoreRule()
216 const auto& orig_axis = src_layout[j]; in GetStoreRule()
217 const IterVar& orig_axis_impl = src_layout->axes[j]; in GetStoreRule()
221 const int32_t factor = src_layout.FactorOf(orig_axis); in GetStoreRule()
267 CHECK_EQ(src_index.size(), self->src_layout->axes.size()) in ForwardIndex()
268 << "Input mismatch with layout " << self->src_layout; in ForwardIndex()
337 return TransformShape(shape, self->src_layout->axes, in ForwardShape()
352 n->src_layout = src_layout; in make()
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/lang/ |
| H A D | data_layout.cc | 204 const Layout& src_layout, in GetStoreRule() argument
206 if (!src_layout.defined() || src_layout.name().empty() || in GetStoreRule()
215 for (size_t j = 0; j < src_layout.ndim(); ++j) { in GetStoreRule()
216 const auto& orig_axis = src_layout[j]; in GetStoreRule()
217 const IterVar& orig_axis_impl = src_layout->axes[j]; in GetStoreRule()
221 const int32_t factor = src_layout.FactorOf(orig_axis); in GetStoreRule()
267 CHECK_EQ(src_index.size(), self->src_layout->axes.size()) in ForwardIndex()
268 << "Input mismatch with layout " << self->src_layout; in ForwardIndex()
337 return TransformShape(shape, self->src_layout->axes, in ForwardShape()
352 n->src_layout = src_layout; in make()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/tir/ir/ |
| H A D | data_layout.cc | 202 inline bool GetStoreRule(Array<PrimExpr>* rule, const Layout& src_layout, in GetStoreRule() argument
204 if (!src_layout.defined() || src_layout.name().empty() || !dst_layout.defined() || in GetStoreRule()
213 for (size_t j = 0; j < src_layout.ndim(); ++j) { in GetStoreRule()
214 const auto& orig_axis = src_layout[j]; in GetStoreRule()
215 const IterVar& orig_axis_impl = src_layout->axes[j]; in GetStoreRule()
219 const int32_t factor = src_layout.FactorOf(orig_axis); in GetStoreRule()
266 CHECK_EQ(src_index.size(), self->src_layout->axes.size()) in ForwardIndex()
267 << "Input mismatch with layout " << self->src_layout; in ForwardIndex()
348 n->src_layout = std::move(src_layout); in BijectiveLayout()
353 if (GetStoreRule(&n->forward_rule, n->src_layout, n->dst_layout)) { in BijectiveLayout()
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/ |
| H A D | util.py | 312 def get_shape(src_shape, src_layout, dst_layout): argument
332 if src_layout == dst_layout:
335 if isinstance(src_layout, str):
336 src_layout = layout(src_layout)
340 assert len(src_layout) == len(dst_layout), \
341 "Incompatible layout %s vs %s" % (src_layout, dst_layout)
343 layout_mapping = bijective_layout(src_layout, dst_layout)
345 tvm.convert([i for i in range(len(src_layout))]))
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/ |
| H A D | util.py | 312 def get_shape(src_shape, src_layout, dst_layout): argument
332 if src_layout == dst_layout:
335 if isinstance(src_layout, str):
336 src_layout = layout(src_layout)
340 assert len(src_layout) == len(dst_layout), \
341 "Incompatible layout %s vs %s" % (src_layout, dst_layout)
343 layout_mapping = bijective_layout(src_layout, dst_layout)
345 tvm.convert([i for i in range(len(src_layout))]))
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/tir/ |
| H A D | data_layout.py | 186 def bijective_layout(src_layout, dst_layout): argument
202 if isinstance(src_layout, str):
203 src_layout = layout(src_layout)
206 return _ffi_api.BijectiveLayout(src_layout, dst_layout)
|
| /dports/sysutils/conan/conan-1.44.0/conans/client/cmd/ |
| H A D | copy.py | 41 src_layout = cache.package_layout(src_ref, short_paths)
42 src_metadata = src_layout.load_metadata()
46 export_origin = src_layout.export()
58 export_sources_origin = src_layout.export_sources()
70 package_path_origin = src_layout.package(pref_origin)
83 src_download = src_layout.download_package(pref_origin)
90 src_download = src_layout.download_export()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/transforms/ |
| H A D | transform_layout.h | 88 Expr Transform(Expr raw, const Layout& src_layout, const Layout& dst_layout) { in Transform() argument
89 if (src_layout.Equals(dst_layout)) { in Transform()
94 std::make_tuple<>(raw.get(), src_layout.name(), dst_layout.name()); in Transform()
101 Expr transform = TransformHelper(raw, src_layout, dst_layout); in Transform()
118 Expr TransformHelper(Expr raw, Layout src_layout, Layout dst_layout) { in TransformHelper() argument
119 if (src_layout.Equals(dst_layout)) { in TransformHelper()
125 Layout new_src_layout = src_layout; in TransformHelper()
126 if (src_layout.ndim_primal() < dst_layout.ndim_primal()) { in TransformHelper()
132 int num_new_axis = dst_layout.ndim_primal() - src_layout.ndim_primal(); in TransformHelper()
133 new_src_layout = src_layout.ExpandPrimal(dst_layout); in TransformHelper()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/ |
| H A D | util.py | 333 def get_shape(src_shape, src_layout, dst_layout): argument
353 if src_layout == dst_layout:
356 if isinstance(src_layout, str):
357 src_layout = layout(src_layout)
361 assert len(src_layout) == len(dst_layout), "Incompatible layout %s vs %s" % (
362 src_layout,
366 layout_mapping = bijective_layout(src_layout, dst_layout)
367 dst_indices = layout_mapping.forward_index(tvm.runtime.convert(list(range(len(src_layout)))))
|
| /dports/multimedia/mencoder/mplayer-export-2021-12-29/libaf/ |
| H A D | reorder_ch.h | 93 int src_layout,
101 int src_layout,
118 int src_layout,
127 int src_layout,
|
| H A D | reorder_ch.c | 281 int src_layout, in reorder_channel_copy() argument
287 if (dest_layout==src_layout) { in reorder_channel_copy()
294 AF_GET_CH_NUM_WITH_LFE(src_layout), in reorder_channel_copy()
298 switch ((src_layout<<16)|dest_layout) { in reorder_channel_copy()
1150 int src_layout, in reorder_channel() argument
1155 if (dest_layout==src_layout) in reorder_channel()
1165 switch ((src_layout<<16)|dest_layout) { in reorder_channel()
1319 int src_layout, in reorder_channel_copy_nch() argument
1327 src_layout < 0 || dest_layout < 0 || in reorder_channel_copy_nch()
1346 int src_layout, in reorder_channel_nch() argument
[all …]
|
| /dports/multimedia/mplayer/mplayer-export-2021-12-29/libaf/ |
| H A D | reorder_ch.h | 93 int src_layout,
101 int src_layout,
118 int src_layout,
127 int src_layout,
|
| H A D | reorder_ch.c | 281 int src_layout, in reorder_channel_copy() argument
287 if (dest_layout==src_layout) { in reorder_channel_copy()
294 AF_GET_CH_NUM_WITH_LFE(src_layout), in reorder_channel_copy()
298 switch ((src_layout<<16)|dest_layout) { in reorder_channel_copy()
1150 int src_layout, in reorder_channel() argument
1155 if (dest_layout==src_layout) in reorder_channel()
1165 switch ((src_layout<<16)|dest_layout) { in reorder_channel()
1319 int src_layout, in reorder_channel_copy_nch() argument
1327 src_layout < 0 || dest_layout < 0 || in reorder_channel_copy_nch()
1346 int src_layout, in reorder_channel_nch() argument
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/relay/pass/ |
| H A D | alter_op_layout.cc | 53 Expr TransformLayout(Expr raw, Layout src_layout, Layout dst_layout) { in TransformLayout() argument
54 if (src_layout.Equals(dst_layout)) { in TransformLayout()
60 Layout new_src_layout = src_layout; in TransformLayout()
61 if (src_layout.ndim_primal() < dst_layout.ndim_primal()) { in TransformLayout()
62 int num_new_axis = dst_layout.ndim_primal() - src_layout.ndim_primal(); in TransformLayout()
63 new_src_layout = src_layout.ExpandPrimal(dst_layout); in TransformLayout()
106 Expr Transform(Expr raw, const Layout& src_layout, const Layout& dst_layout) { in Transform() argument
107 if (src_layout.Equals(dst_layout)) { return raw; } in Transform()
110 std::make_tuple<>(raw.get(), src_layout.name(), dst_layout.name()); in Transform()
117 Expr transform = TransformLayout(raw, src_layout, dst_layout); in Transform()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/relay/pass/ |
| H A D | alter_op_layout.cc | 53 Expr TransformLayout(Expr raw, Layout src_layout, Layout dst_layout) { in TransformLayout() argument
54 if (src_layout.Equals(dst_layout)) { in TransformLayout()
60 Layout new_src_layout = src_layout; in TransformLayout()
61 if (src_layout.ndim_primal() < dst_layout.ndim_primal()) { in TransformLayout()
62 int num_new_axis = dst_layout.ndim_primal() - src_layout.ndim_primal(); in TransformLayout()
63 new_src_layout = src_layout.ExpandPrimal(dst_layout); in TransformLayout()
106 Expr Transform(Expr raw, const Layout& src_layout, const Layout& dst_layout) { in Transform() argument
107 if (src_layout.Equals(dst_layout)) { return raw; } in Transform()
110 std::make_tuple<>(raw.get(), src_layout.name(), dst_layout.name()); in Transform()
117 Expr transform = TransformLayout(raw, src_layout, dst_layout); in Transform()
|
| /dports/lang/rust/rustc-1.58.1-src/compiler/rustc_codegen_cranelift/src/ |
| H A D | unsize.rs | 61 src_layout: TyAndLayout<'tcx>, in unsize_ptr()
65 match (&src_layout.ty.kind(), &dst_layout.ty.kind()) { in unsize_ptr()
72 let (a, b) = (src_layout.ty.boxed_ty(), dst_layout.ty.boxed_ty()); in unsize_ptr()
78 if src_layout == dst_layout { in unsize_ptr()
83 for i in 0..src_layout.fields.count() { in unsize_ptr()
84 let src_f = src_layout.field(fx, i); in unsize_ptr()
85 assert_eq!(src_layout.fields.offset(i).bytes(), 0); in unsize_ptr()
90 assert_eq!(src_layout.size, src_f.size); in unsize_ptr()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/driver/tvmc/ |
| H A D | test_common.py | 42 and node.attrs.src_layout == "NHWC"
66 and node.attrs.src_layout == "NCHW"
90 and node.attrs.src_layout == "NHWC"
114 and node.attrs.src_layout == "NCHW"
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/tests/python/ |
| H A D | test_topi_util.py | 22 def verify_get_shape(src_shape, src_layout, dst_layout, expect_shape): argument
23 dst_shape = topi.util.get_shape(src_shape, src_layout, dst_layout)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/tests/python/ |
| H A D | test_topi_util.py | 22 def verify_get_shape(src_shape, src_layout, dst_layout, expect_shape): argument
23 dst_shape = topi.util.get_shape(src_shape, src_layout, dst_layout)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/topi/python/ |
| H A D | test_topi_util.py | 22 def verify_get_shape(src_shape, src_layout, dst_layout, expect_shape): argument
23 dst_shape = topi.util.get_shape(src_shape, src_layout, dst_layout)
|