| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | random_choice_with_mask_impl.cu | 102 __global__ void Reshape2Index(const int input_size, const int input_shape_size, const int d1, const… in Reshape2Index() argument
113 index_pos = pos * input_shape_size; in Reshape2Index()
115 for (int i = 0; i < input_shape_size; i++) { in Reshape2Index()
119 for (int i = MAX_DIMENSION - input_shape_size; i < MAX_DIMENSION; i++) { in Reshape2Index()
184 __global__ void MoveToOutput(const int input_shape_size, const int count, const T *input, S *output… in MoveToOutput() argument
194 pos *= input_shape_size; in MoveToOutput()
195 idx *= input_shape_size; in MoveToOutput()
196 for (size_t j = 0; j < input_shape_size; j++) { in MoveToOutput()
208 pos *= input_shape_size; in MoveToOutput()
209 idx *= input_shape_size; in MoveToOutput()
[all …]
|
| D | random_choice_with_mask_impl.cuh | 31 void CalRandomChoiceWithMask(const int &input_size, const int &input_shape_size, const int &d1, con…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | batch_to_space_infer.c | 22 size_t input_shape_size = 0; in SetOutputShapeFromParam() local
23 ShapeSet(input_shape, &input_shape_size, inputs[0]->shape_, inputs[0]->shape_size_); in SetOutputShapeFromParam()
25 if (input_shape_size != 4) { in SetOutputShapeFromParam()
56 size_t output_shape_size = input_shape_size; in SetOutputShapeFromParam()
67 size_t input_shape_size = 0; in SetOutputShapeFromInput() local
68 ShapeSet(input_shape, &input_shape_size, inputs[0]->shape_, inputs[0]->shape_size_); in SetOutputShapeFromInput()
69 if (input_shape_size != 4) { in SetOutputShapeFromInput()
104 size_t output_shape_size = input_shape_size; in SetOutputShapeFromInput()
|
| D | flatten_infer.c | 39 size_t input_shape_size = 0; in FlattenInferShape() local
40 ShapeSet(input_shape, &input_shape_size, input->shape_, input->shape_size_); in FlattenInferShape()
44 for (size_t i = 1; i < input_shape_size; i++) { in FlattenInferShape()
|
| D | depth_to_space_infer.c | 40 size_t input_shape_size = 0; in DepthToSpaceInferShape() local
41 ShapeSet(input_shape, &input_shape_size, input->shape_, input->shape_size_); in DepthToSpaceInferShape()
48 size_t output_shape_size = input_shape_size; in DepthToSpaceInferShape()
|
| D | argmin_max_infer.c | 59 int input_shape_size = (int)input->shape_size_; in ArgMinMaxInferShape() local
60 int axis = param->axis_ < 0 ? param->axis_ + input_shape_size : param->axis_; in ArgMinMaxInferShape()
61 if (axis >= input_shape_size || axis < 0) { in ArgMinMaxInferShape()
|
| D | pooling_infer.c | 85 size_t input_shape_size = 0; in PoolingInferShape() local
86 ShapeSet(input_shape, &input_shape_size, input->shape_, input->shape_size_); in PoolingInferShape()
89 SetShapeArray(output, input_shape, input_shape_size); in PoolingInferShape()
|
| D | broadcast_to_infer.c | 157 int input_shape_size = input->shape_size_; in BroadcastToInferShape() local
159 int ndim = input_shape_size; in BroadcastToInferShape()
186 …if (BroadCastToShape(input_shape_size, dst_shape_size, input_shape, dst_shape, &ndim, output_shape, in BroadcastToInferShape()
|
| D | common_infer.c | 407 size_t input_shape_size = 0; in FftInferShape() local
408 ShapeSet(input_shape, &input_shape_size, input->shape_, input->shape_size_); in FftInferShape()
409 if (input_shape_size == 0) { in FftInferShape()
412 input_shape_size--; in FftInferShape()
413 SetShapeArray(output, input_shape, input_shape_size); in FftInferShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | broadcast_to_cpu_kernel.cc | 33 size_t input_shape_size = input_shape_.size(); in InitKernel() local
35 if (output_shape_size < input_shape_size) { in InitKernel()
43 size_t offset = output_shape_size - input_shape_size; in InitKernel()
44 for (size_t i = 0; i < input_shape_size; ++i) { in InitKernel()
51 for (size_t i = 0; i < input_shape_size; ++i) { in InitKernel()
57 shape_info_.input_shape_size_ = SizeToInt(input_shape_size); in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_ascend.cc | 95 auto input_shape_size = AnfAlgo::GetPrevNodeOutputInferShape(cnode, index).size(); in GetPriorityMatchFormat() local
96 if (input_shape_size == k5dSize) { in GetPriorityMatchFormat()
99 need_change_nd = (need_change_nd || (input_shape_size != k4dSize && input_shape_size > 1)); in GetPriorityMatchFormat()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 51 const size_t input_shape_size = 4; in InferImplPooling() local
52 if (input_shape->shape().size() != input_shape_size) { in InferImplPooling()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_model_parser.cc | 335 for (int i = 0; i < caffe_model_.input_shape_size(); i++) { in ConvertGraphInputsOfShape()
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1779 size_t input_shape_size = input_shape->shape().size(); in ExportMergeLayerNorm() local
1780 for (size_t i = 0; i < input_shape_size - 1; i++) { in ExportMergeLayerNorm()
1786 new_input_shape.push_back(input_shape->shape()[input_shape_size - kOneNum]); in ExportMergeLayerNorm()
|