Searched defs:softmax_info (Results 1 – 1 of 1) sorted by relevance
74 SoftmaxKernelInfo softmax_info{ beta, IS_LOG, input->info()->data_type(), axis }; in configure() local84 SoftmaxKernelInfo softmax_info{ beta, IS_LOG, input->data_type(), axis }; in validate() local