Home
last modified time | relevance | path

Searched defs:out_shape_ (Results 1 – 16 of 16) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dspace_to_depth.h40 GpuTensorInfo out_shape_; variable
Done_hot.h45 GpuTensorInfo out_shape_; variable
Dsoftmax.h44 GpuTensorInfo out_shape_; variable
Dstack.h50 cl_int4 out_shape_ = {}; variable
Darithmetic.h47 GpuTensorInfo out_shape_; variable
Dsplit.h44 cl_int4 out_shape_ = {}; variable
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/int8/
Darithmetic_int8.h45 GpuTensorInfo out_shape_; variable
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dtranspose_fp32.h54 int *out_shape_ = nullptr; variable
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dtranspose_int8.h54 int out_shape_[20] = {0}; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dcrop_parameter.h32 int *out_shape_; member
Dsigmoid_parameter.h28 const int *out_shape_; member
Dsqueeze_parameter.h37 const int *out_shape_; member
Dunsqueeze_parameter.h39 const int *out_shape_; member
Darithmetic.h35 int out_shape_[ARITHMETIC_SUPPORT_DIMS_NUM]; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dtile_base.h32 int out_shape_[MAX_TILE_DIM_SIZE]; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dquantize.h163 int *out_shape_; member