Searched refs:TopkParameter (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | topk_int8.cc | 37 TopkParameter *parameter = reinterpret_cast<TopkParameter *>(op_parameter_); in ReSize()
57 TopkParameter *parameter = reinterpret_cast<TopkParameter *>(op_parameter_); in Run()
63 TopkInt8(input_data, output_data, output_index, reinterpret_cast<TopkParameter *>(op_parameter_)); in Run()
|
| D | topk_int8.h | 29 TopkParameter *param = reinterpret_cast<TopkParameter *>(op_parameter_); in TopKInt8CPUKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | topk_fp32.h | 27 typedef struct TopkParameter { struct
37 } TopkParameter; argument
42 void Topk(float *input_data, float *output_data, int32_t *output_index, TopkParameter *parameter);
|
| D | topk_fp32.c | 41 void Topk(float *input_data, float *output_data, int32_t *output_index, TopkParameter *parameter) { in Topk()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/ |
| D | topk_populate.cc | 31 auto *param = reinterpret_cast<TopkParameter *>(malloc(sizeof(TopkParameter))); in PopulateTopKParameter()
36 memset(param, 0, sizeof(TopkParameter)); in PopulateTopKParameter()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/v0/ |
| D | topk_populate_v0.cc | 32 auto *topk_param = reinterpret_cast<TopkParameter *>(malloc(sizeof(TopkParameter))); in PopulateTopKParameter()
37 memset(topk_param, 0, sizeof(TopkParameter)); in PopulateTopKParameter()
|
| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | topk_infer_test.cc | 38 TopkParameter *parameter = new TopkParameter; in TEST_F()
75 TopkParameter *parameter = new TopkParameter; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | topk_fp32.h | 29 topk_param_ = reinterpret_cast<TopkParameter *>(op_parameter_); in TopKCPUKernel()
38 TopkParameter *topk_param_;
|
| D | topk_fp32.cc | 71 Topk(input_data, output_data, output_index, reinterpret_cast<TopkParameter *>(op_parameter_)); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | topk_infer.c | 44 TopkParameter *param = (TopkParameter *)parameter; in TopKInferShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | topk_int8.h | 31 void TopkInt8(int8_t *input_data, int8_t *output_data, int32_t *output_index, TopkParameter *parame…
|
| D | topk_int8.c | 27 void TopkInt8(int8_t *input_data, int8_t *output_data, int32_t *output_index, TopkParameter *parame… in TopkInt8()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | topk_int8_tests.cc | 43 TopkParameter parameter = {{}, 2, true, 3, 4}; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | topk_fp32_tests.cc | 43 TopkParameter parameter = {{}, 2, true, 3, 4}; in TEST_F()
|