Searched defs:out_dim (Results 1 – 5 of 5) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
D | redistribution_operator_infer.cc | 102 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/ |
D | cumsum_fp32.cc | 97 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
|
D | affine_fp32.cc | 246 int out_dim = affine_parameter_->output_dim_; in FullMatmulKernelCreate() local
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
D | cumsum_fp32.c | 22 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/ |
D | splice_infer.c | 41 int out_dim = param->output_dim_; in SpliceInferShape() local
|