Searched refs:in_shape0 (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | broadcast_to_infer.c | 67 const int *input_shape1, int *ndim, int *in_shape0, int *in_shape1) { in MakeUpInputShapes() argument
74 in_shape0[i] = 1; in MakeUpInputShapes()
76 in_shape0[i] = input_shape0[j++]; in MakeUpInputShapes()
90 in_shape0[i] = input_shape0[i]; in MakeUpInputShapes()
95 in_shape0[i] = input_shape0[i]; in MakeUpInputShapes()
100 int BroadCastOutputShape(const int *in_shape0, const int *in_shape1, const int ndim, int *out_shape, in BroadCastOutputShape() argument
103 if (in_shape0[i] != in_shape1[i]) { in BroadCastOutputShape()
104 if (in_shape0[i] == 1) { in BroadCastOutputShape()
107 out_shape[i] = in_shape0[i]; in BroadCastOutputShape()
113 out_shape[i] = in_shape0[i]; in BroadCastOutputShape()
[all …]
|
| D | arithmetic_infer.c | 39 … const int *input_shape1, int *ndim, int *in_shape0, int *in_shape1, int *out_shape, in BroadCastInferShape() argument
44 …apes(input_shape0_size, input_shape1_size, input_shape0, input_shape1, ndim, in_shape0, in_shape1); in BroadCastInferShape()
49 return BroadCastOutputShape(in_shape0, in_shape1, *ndim, out_shape, has_broad_cast); in BroadCastInferShape()
77 int in_shape0[MAX_SHAPE_SIZE] = {0}; in ArithmeticInferShape() local
82 …hape(input_shape0_size, input_shape1_size, input_shape0, input_shape1, &ndim, in_shape0, in_shape1, in ArithmeticInferShape()
94 memcpy(param->in_shape0_, in_shape0, ndim * sizeof(int)); in ArithmeticInferShape()
|
| D | arithmetic_grad_infer.c | 44 int in_shape0[MAX_SHAPE_SIZE] = {0}; in ArithmeticGradInferShape() local
46 ShapeSet(in_shape0, &in_shape0_size, x1->shape_, x1->shape_size_); in ArithmeticGradInferShape()
67 param->in_shape1_[i] = in_shape0[j++]; in ArithmeticGradInferShape()
86 param->in_shape0_[i] = in_shape0[i]; in ArithmeticGradInferShape()
93 param->in_shape0_[i] = in_shape0[i]; in ArithmeticGradInferShape()
|
| D | broadcast_to_infer.h | 29 const int *input_shape1, int *ndim, int *in_shape0, int *in_shape1);
30 int BroadCastOutputShape(const int *in_shape0, const int *in_shape1, const int ndim, int *out_shape,
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | merge.cc | 41 std::vector<int64_t> in_shape0 = inputs_shape[0]->cast<abstract::ShapePtr>()->shape(); in MergeInfer() local
43 auto output1 = std::make_shared<abstract::AbstractTensor>(infered_type, in_shape0); in MergeInfer()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | reduce_fp32.c | 393 int *in_shape0 = in_shape[0]; in ReduceInferShape() local
449 out_shape[j++] = in_shape0[i]; in ReduceInferShape()
|