Home
last modified time | relevance | path

Searched defs:axis_ (Results 1 – 25 of 56) sorted by relevance

123

/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dgather.h44 int axis_ = {0}; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dunsqueeze_parameter.h25 int axis_; member
41 int64_t axis_; member
Dgather_parameter.h25 int axis_; member
Dstack_parameter.h24 int32_t axis_; member
Dglu_parameter.h23 int axis_; member
Dcumsum_parameter.h26 int axis_; member
Dsoftmax_parameter.h25 int32_t axis_; member
Done_hot_parameter.h24 int axis_; member
Dunstack_parameter.h26 int axis_; member
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/kernels/
Dsliding_window_op.h60 int32_t axis_; // The axis along which sliding window is computed, only support 0/-1 for now. variable
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dglu_fusion.h43 mutable VarPtr axis_ = nullptr; variable
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dgather_fp32.h41 int axis_ = 0; variable
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dgather_npu.h47 int axis_; variable
Dunsqueeze_npu.h47 vector<int> axis_; variable
Dconcat_npu.h56 int axis_ = 0; variable
Dsplit_npu.h53 int axis_ = 0; variable
Dscale_npu.h54 int axis_ = 0; variable
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconcat_int8_coder.h53 int axis_ = 0; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dreverse_fp32.h29 int axis_[REVERSE_SHAPE_MAX_SIZE]; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dreduce_cpu_kernel.h43 std::vector<int64_t> axis_; variable
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Done_hot_base.h43 int axis_ = 0; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dgather_v2_info.h67 int64_t axis_; variable
Donehot_info.h58 int64_t axis_ = -1; variable
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dconcat_int8_wrapper.h28 int axis_; member
/third_party/boost/boost/histogram/axis/
Diterator.hpp30 const Axis& axis_; member in boost::histogram::axis::iterator

123