Home
last modified time | relevance | path

Searched defs:out_dim (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dredistribution_operator_infer.cc102 int64_t out_dim = out_tensor_map_.GetDimByIdx(index); in InferSplitByAxis() local
127 int64_t out_dim = out_tensor_map_.GetDimByIdx(index); in InferPermuteByAxis() local
170 int64_t out_dim = out_tensor_map_.GetDimByIdx(index); in InferConcatByAxis() local
245 int64_t out_dim = out_tensor_map_.GetDimByIdx(index); in TransferPermuteByAxis() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dcumsum_fp32.cc97 int out_dim = MSMIN(out_dim_ - unit_ * task_id, unit_); in DoCumsum() local
114 int out_dim = MSMIN(out_dim_ - unit_ * task_id, unit_); in DoCumsumInt() local
Daffine_fp32.cc246 int out_dim = affine_parameter_->output_dim_; in FullMatmulKernelCreate() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dcumsum_fp32.c22 void Cumsum(const float *input, float *output, int out_dim, int axis_dim, int inner_dim, bool exclu… in Cumsum()
84 void CumsumReverse(const float *input, float *output, int out_dim, int axis_dim, int inner_dim, boo… in CumsumReverse()
143 void CumsumInt(const int *input, int *output, int out_dim, int axis_dim, int inner_dim, bool exclus… in CumsumInt()
204 void CumsumReverseInt(const int *input, int *output, int out_dim, int axis_dim, int inner_dim, bool… in CumsumReverseInt()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dsplice_infer.c41 int out_dim = param->output_dim_; in SpliceInferShape() local