Searched refs:SoftMaxInferShape (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | softmax_infer.c | 20 int SoftMaxInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **outputs, size_t o… in SoftMaxInferShape() function
42 REG_INFER(Softmax, PrimType_Softmax, SoftMaxInferShape)
|
| D | softmax_infer.h | 26 int SoftMaxInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **outputs, size_t o…
|
| D | infer_register.c | 283 g_infer_func[PrimType_Softmax] = SoftMaxInferShape; in RegAllInferFunc2()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | softmax.cc | 48 abstract::ShapePtr SoftMaxInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBaseP… in SoftMaxInferShape() function
82 …return abstract::MakeAbstract(SoftMaxInferShape(primitive, input_args), SoftMaxInferType(primitive… in SoftmaxInfer()
|
| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | softmax_infer_test.cc | 38 …int ret = SoftMaxInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), output… in TEST_F()
|