| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_softmax.cc | 38 EXPECT_EQ(softmax->get_axis(), init_data); in TEST_F()
40 EXPECT_EQ(softmax->get_axis(), init_data); in TEST_F()
68 EXPECT_EQ(softmax->get_axis(), init_data); in TEST_F()
70 EXPECT_EQ(softmax->get_axis(), init_data); in TEST_F()
98 EXPECT_EQ(softmax->get_axis(), init_data); in TEST_F()
100 EXPECT_EQ(softmax->get_axis(), init_data); in TEST_F()
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_auto_monad_momentum_loss.py | 44 def get_axis(x): function
61 return self.reduce_mean(self.square(diff), get_axis(diff))
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | squeeze_fusion.cc | 98 if (squeeze_prim->get_axis() == unsqueeze_prim->get_axis()) { in Process()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | glu.cc | 26 int64_t GLU::get_axis() const { in get_axis() function in mindspore::ops::GLU
|
| D | scale.cc | 24 int64_t Scale::get_axis() const { return GetValue<int64_t>(GetAttr(kAxis)); } in get_axis() function in mindspore::ops::Scale
|
| D | log_softmax.h | 46 int64_t get_axis() const;
|
| D | one_hot.h | 45 int64_t get_axis() const;
|
| D | scale.h | 60 int64_t get_axis() const;
|
| D | glu.h | 36 int64_t get_axis() const;
|
| D | lp_normalization.h | 58 int64_t get_axis() const;
|
| D | unstack.h | 49 int64_t get_axis() const;
|
| D | unpack.h | 49 int64_t get_axis() const;
|
| D | concat.h | 46 int64_t get_axis() const;
|
| D | stack.h | 49 int64_t get_axis() const;
|
| D | pack.h | 50 int64_t get_axis() const;
|
| D | lp_normalization.cc | 30 int64_t LpNormalization::get_axis() const { in get_axis() function in mindspore::ops::LpNormalization
|
| D | softmax.h | 46 std::vector<int64_t> get_axis() const;
|
| D | squeeze.h | 50 std::vector<int64_t> get_axis() const;
|
| D | reverse_v2.h | 46 std::vector<int64_t> get_axis() const;
|
| D | unsqueeze.h | 54 std::vector<int64_t> get_axis() const;
|
| D | crop.h | 59 int64_t get_axis() const;
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_auto_parallel_flag.py | 77 def get_axis(x): function
94 return self.reduce_mean(self.square(diff), get_axis(diff))
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | topk_fusion.h | 61 int64_t get_axis() const;
|
| /third_party/skia/third_party/externals/harfbuzz/src/ |
| D | hb-ot-layout-base-table.hh | 448 const Axis &get_axis (hb_direction_t direction) const in get_axis() function
462 …if (unlikely (!get_axis (direction).get_baseline (baseline_tag, script_tag, language_tag, &base_co… in get_baseline()
482 if (!get_axis (direction).get_min_max (script_tag, language_tag, feature_tag, in get_min_max()
|
| /third_party/harfbuzz/src/ |
| D | hb-ot-layout-base-table.hh | 448 const Axis &get_axis (hb_direction_t direction) const in get_axis() function
462 …if (unlikely (!get_axis (direction).get_baseline (baseline_tag, script_tag, language_tag, &base_co… in get_baseline()
482 if (!get_axis (direction).get_min_max (script_tag, language_tag, feature_tag, in get_min_max()
|