Home
last modified time | relevance | path

Searched defs:input_shape1_size (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dwhere_infer.c22 int WhereBroadCastInferShape(const int input_shape0_size, const int input_shape1_size, const int *i… in WhereBroadCastInferShape()
68 size_t input_shape1_size = input1->shape_size_; in WhereInferShape() local
Darithmetic_infer.c38 int BroadCastInferShape(const int input_shape0_size, const int input_shape1_size, const int *input_… in BroadCastInferShape()
70 size_t input_shape1_size = input1->shape_size_; in ArithmeticInferShape() local
Dbroadcast_to_infer.c67 void MakeUpInputShapes(const int input_shape0_size, const int input_shape1_size, const int *input_s… in MakeUpInputShapes()
120 int BroadCastToShape(const int input_shape0_size, const int input_shape1_size, const int *input_sha… in BroadCastToShape()