| /third_party/boost/boost/histogram/axis/ |
| D | traits.hpp | 123 namespace axis { namespace
265 constexpr unsigned options(const Axis& axis) noexcept { in options()
272 unsigned options(const variant<Ts...>& axis) noexcept { in options()
283 constexpr bool inclusive(const Axis& axis) noexcept { in inclusive()
290 bool inclusive(const variant<Ts...>& axis) noexcept { in inclusive()
301 constexpr bool ordered(const Axis& axis) noexcept { in ordered()
308 bool ordered(const variant<Ts...>& axis) noexcept { in ordered()
319 constexpr bool continuous(const Axis& axis) noexcept { in continuous()
326 bool continuous(const variant<Ts...>& axis) noexcept { in continuous()
335 index_type extent(const Axis& axis) noexcept { in extent()
[all …]
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | softmax_tests.cc | 25 OpParameter *CreateParameter(int axis) { in CreateParameter()
33 int axis = 1; variable
47 int axis = 3; variable
61 int axis = 1; variable
75 int axis = 1; variable
90 int axis = 3; variable
105 int axis = 1; variable
|
| D | one_hot_tests.cc | 25 OpParameter *CreateParameter(int axis) { in CreateParameter()
34 int axis = -1; in TEST_F() local
55 int axis = -1; in TEST_F() local
76 int axis = -1; in TEST_F() local
99 int axis = -1; in TEST_F() local
122 int axis = 2; in TEST_F() local
143 int axis = 2; in TEST_F() local
165 int axis = 2; in TEST_F() local
186 int axis = 2; in TEST_F() local
209 int axis = 1; in TEST_F() local
[all …]
|
| D | argminmax_tests.cc | 26 OpParameter *CreateParameter(schema::PrimitiveType type, int axis, int topk, bool out_value, bool k… in CreateParameter()
40 int axis = 0; in TEST_F() local
55 int axis = 0; in TEST_F() local
70 int axis = 1; in TEST_F() local
86 int axis = 1; in TEST_F() local
103 int axis = 2; in TEST_F() local
120 int axis = 2; in TEST_F() local
138 int axis = 2; in TEST_F() local
156 int axis = 3; in TEST_F() local
173 int axis = 3; in TEST_F() local
[all …]
|
| D | reduce_tests.cc | 26 OpParameter *CreateParameter(const std::vector<int> &axis, schema::ReduceMode mode, bool keep_dims)… in CreateParameter()
41 std::vector<int> axis = {1, 2}; in TEST_F() local
56 std::vector<int> axis = {1, 2}; in TEST_F() local
71 std::vector<int> axis = {2, 3}; in TEST_F() local
86 std::vector<int> axis = {2, 3}; in TEST_F() local
101 std::vector<int> axis = {3}; in TEST_F() local
|
| D | concat_tests.cc | 25 OpParameter *CreateParameter(int axis) { in CreateParameter()
36 int axis = 0; in TEST_F() local
51 int axis = 0; in TEST_F() local
66 int axis = 1; in TEST_F() local
98 int axis = 1; in TEST_F() local
193 int axis = 2; in TEST_F() local
220 int axis = 3; in TEST_F() local
252 int axis = 1; in TEST_F() local
271 int axis = -1; in TEST_F() local
295 int axis = 3; in TEST_F() local
[all …]
|
| D | transpose_tests.cc | 41 for (int axis : perm) { in TEST_F() local
59 for (int axis : perm) { in TEST_F() local
77 for (int axis : perm) { in TEST_F() local
95 for (int axis : perm) { in TEST_F() local
113 for (int axis : perm) { in TEST_F() local
|
| D | stack_tests.cc | 25 OpParameter *CreateParameter(int axis) { in CreateParameter()
35 int axis = 1; in TEST_F() local
52 int axis = 1; in TEST_F() local
71 int axis = 1; in TEST_F() local
90 int axis = 2; in TEST_F() local
109 int axis = 2; in TEST_F() local
128 int axis = 3; in TEST_F() local
|
| D | scale_tests.cc | 25 OpParameter *CreateParameter(int axis, int activation_type = schema::ActivationType_NO_ACTIVATION) { in CreateParameter()
34 int axis = 1; in TEST_F() local
52 int axis = 3; in TEST_F() local
70 int axis = 3; in TEST_F() local
|
| /third_party/mindspore/mindspore/numpy/ |
| D | array_ops.py | 40 def expand_dims(a, axis): argument
80 def squeeze(a, axis=None): argument
140 def rollaxis(x, axis, start=0): argument
327 def _move_axes_for_concatenate(arr_shape, axis): argument
368 def concatenate(arrays, axis=0): argument
437 def append(arr, values, axis=None): argument
876 def stack(arrays, axis=0): argument
998 def roll_along_axis(a, shift, axis): argument
1038 def roll(a, shift, axis=None): argument
1319 def array_split(x, indices_or_sections, axis=0): argument
[all …]
|
| D | math_ops.py | 107 def count_nonzero(x, axis=None, keepdims=False): argument
582 def mean(a, axis=None, keepdims=False, dtype=None): argument
873 def std(x, axis=None, ddof=0, keepdims=False): argument
916 def var(x, axis=None, ddof=0, keepdims=False): argument
957 def ptp(x, axis=None, keepdims=False): argument
995 def average(x, axis=None, weights=None, returned=False): argument
1057 def compute_weights_for_mean(x, x_avg, axis): argument
1075 def comput_avg(x, axis, weights): argument
1382 def amax(a, axis=None, keepdims=False, initial=None, where=True): argument
1437 def amin(a, axis=None, keepdims=False, initial=None, where=True): argument
[all …]
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_squeeze.py | 24 def __init__(self, axis): argument
32 def get_output(x, axis=(), enable_graph_kernel=False): argument
39 def test_squeeze(shape, dtype, axis=()): argument
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | unpack.cc | 21 void Unpack::Init(const int64_t axis) { this->set_axis(axis); } in Init()
22 void Unpack::set_axis(const int64_t axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis()
34 int64_t axis = GetValue<int64_t>(primitive->GetAttr(kAxis)); in UnpackInfer() local
|
| D | unstack.cc | 21 void Unstack::Init(const int64_t axis) { this->set_axis(axis); } in Init()
22 void Unstack::set_axis(const int64_t axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis()
31 int64_t axis = GetValue<int64_t>(primitive->GetAttr(kAxis)); in UnstackInfer() local
|
| D | l2_normalize.cc | 23 void L2Normalize::Init(const std::vector<int64_t> &axis, const float epsilon) { in Init()
28 void L2Normalize::set_axis(const std::vector<int64_t> &axis) { (void)AddAttr(kAxis, MakeValue(axis)… in set_axis()
52 for (auto &axis : axiss) { in L2NormalizeInfer() local
|
| D | arg_min.cc | 22 void ArgMin::Init(const int64_t axis, const TypeId output_type) { in Init()
27 void ArgMin::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis()
44 auto axis = GetValue<int64_t>(primitive->GetAttr(kAxis)); in ArgMinInfer() local
|
| D | scale.cc | 21 void Scale::Init(const int64_t axis) { set_axis(axis); } in Init()
23 void Scale::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis()
|
| D | glu.cc | 22 void GLU::Init(int64_t axis) { set_axis(axis); } in Init()
24 void GLU::set_axis(int64_t axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis()
|
| D | log_softmax.cc | 29 void LogSoftmax::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis()
33 void LogSoftmax::Init(const int64_t axis) { this->set_axis(axis); } in Init()
38 auto axis = GetValue<int64_t>(primitive->GetAttr(kAxis)); in LogSoftmaxInferShape() local
|
| D | pack.cc | 22 …_t> _get_pack_shape(std::vector<BaseShapePtr> x_shapes, std::vector<TypePtr> x_types, int64_t axis, in _get_pack_shape()
50 void Pack::set_axis(const int64_t &axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis()
54 void Pack::Init(const int64_t &axis) { this->set_axis(axis); } in Init()
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_shift_op.py | 34 def __init__(self, periods=1, axis=-1): argument
57 def __init__(self, periods=1, axis=-1): argument
115 def test_no_shift(fill_value, dtype, axis): argument
152 def test_2d(fill_value, dtype, axis, periods): argument
168 def test_4d(fill_value, dtype, axis, periods): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | softmax_gpu_kernel.h | 127 auto axis = LongToInt(GetAttr<int64_t>(kernel_node, "axis")); in Init() local
131 std::vector<int> axis; in Init() local
183 void InitSizeByAxis(const std::vector<size_t> &input_shape, const int &axis) { in InitSizeByAxis()
200 void InitSizeByAxis2D(const std::vector<size_t> &input_shape, const int &axis) { in InitSizeByAxis2D()
230 void InitSizeByAxisLastDim(const std::vector<size_t> &input_shape, const int &axis) { in InitSizeByAxisLastDim()
256 void InitSizeByAxisND(const std::vector<size_t> &input_shape, const int &axis) { in InitSizeByAxisND()
|
| /third_party/boost/libs/qvm/test/ |
| D | rot_quat_test.cpp | 19 test_qvm::vector<V1,3> axis; axis.a[0]=1; in test_x() local
43 test_qvm::vector<V1,3> axis; axis.a[1]=1; in test_y() local
67 test_qvm::vector<V1,3> axis; axis.a[2]=1; in test_z() local
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | expand_dims.py | 31 def infer_shape(shape, axis): argument
33 def insert_axis(shape, axis): argument
|
| D | squeeze.py | 31 def infer_shape(shape, axis): argument
33 def squeeze_axis(shape, axis): argument
|