Searched defs:softmax_elements_stride (Results 1 – 1 of 1) sorted by relevance
303 …rward(output_t *dst, const input_t *src, int softmax_elements, int softmax_elements_stride, int ba… in dispatch_softmax_forward()353 …st input_t *grad, const input_t *output, int softmax_elements, int softmax_elements_stride, int ba… in dispatch_softmax_backward()