MCPcopy Create free account
hub / github.com/davisking/dlib / softmax_all_gradient

Function softmax_all_gradient

dlib/cuda/cpu_dlib.cpp:1807–1816  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1805 }
1806
1807 void softmax_all_gradient (
1808 tensor& grad,
1809 const tensor& dest,
1810 const tensor& gradient_input
1811 )
1812 {
1813 DLIB_CASSERT(have_same_dimensions(grad,dest));
1814 DLIB_CASSERT(have_same_dimensions(grad,gradient_input));
1815 ttimpl::softmax_gradient(1, grad.nr()*grad.nc()*grad.k(), grad, dest, gradient_input);
1816 }
1817
1818 // ------------------------------------------------------------------------------------
1819

Callers 2

test_softmax_allFunction · 0.50
backward_inplaceMethod · 0.50

Calls 5

have_same_dimensionsFunction · 0.70
softmax_gradientFunction · 0.70
nrMethod · 0.45
ncMethod · 0.45
kMethod · 0.45

Tested by 1

test_softmax_allFunction · 0.40