Searched refs:axis_dim_ (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | cumsum_fp32.cc | 80 axis_dim_ = input_tensor->shape().at(param_->axis_); in ReSize()
96 float *input = input_data + task_id * unit_ * axis_dim_ * in_dim_; in DoCumsum()
98 float *output = output_data + task_id * unit_ * axis_dim_ * in_dim_; in DoCumsum()
100 Cumsum(input, output, out_dim, axis_dim_, in_dim_, param_->exclusive_); in DoCumsum()
102 CumsumReverse(input, output, out_dim, axis_dim_, in_dim_, param_->exclusive_); in DoCumsum()
113 int *input = input_data + task_id * unit_ * axis_dim_ * in_dim_; in DoCumsumInt()
115 int *output = output_data + task_id * unit_ * axis_dim_ * in_dim_; in DoCumsumInt()
117 CumsumInt(input, output, out_dim, axis_dim_, in_dim_, param_->exclusive_); in DoCumsumInt()
119 CumsumReverseInt(input, output, out_dim, axis_dim_, in_dim_, param_->exclusive_); in DoCumsumInt()
|
| D | unstack_fp32.cc | 45 para->axis_dim_ = 1; in ReSize()
57 para->axis_dim_ = input->DimensionSize(i); in ReSize()
|
| D | cumsum_fp32.h | 42 int axis_dim_ = 1; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | unpack_cpu_kernel.cc | 41 unstack_param_.axis_dim_ = 1; in InitKernel()
50 unstack_param_.axis_dim_ = SizeToInt(input_shape[i]); in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | unstack_parameter.h | 30 int axis_dim_; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | unstack_base.c | 28 int in_offset = i * para->axis_dim_ * para->after_dims_ + j * para->after_dims_; in Unstack()
|