Home
last modified time | relevance | path

Searched refs:softmax_grad (Results 1 – 7 of 7) sorted by relevance

/dports/math/py-theano/Theano-1.0.5/theano/tensor/nnet/tests/
H A Dtest_nnet.py718 if node.op == softmax_grad:
758 if node.op == softmax_grad:
818 assert softmax_grad not in ops
857 assert softmax_grad not in ops
901 assert softmax_grad not in ops
940 assert softmax_grad not in ops
988 assert softmax_grad not in ops
1033 assert softmax_grad not in ops
1090 assert softmax_grad not in ops
1255 if node.op == softmax_grad:
[all …]
/dports/math/py-theano/Theano-1.0.5/theano/tensor/nnet/
H A D__init__.py20 prepend_scalar_to_each_row, relu, softmax, softmax_grad, softmax_graph,
H A Dnnet.py110 dx = softmax_grad(g_sm, outputs[0])
408 softmax_grad = SoftmaxGrad() variable
449 return [softmax_grad(g_sm, outputs[0])]
1567 @gof.local_optimizer([softmax_grad])
1569 if node.op == softmax_grad:
1720 @gof.local_optimizer([softmax_grad])
1722 if not (node.op == softmax_grad):
/dports/math/py-theano/Theano-1.0.5/theano/gpuarray/c_code/
H A Ddnn_softmax_grad.c46 int APPLY_SPECIFIC(softmax_grad)(PyGpuArrayObject *dy, in APPLY_SPECIFIC() argument
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/relay/op/
H A D_tensor_grad.py365 def softmax_grad(orig, grad): function
376 return softmax_grad(sm, grad)
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/relay/op/
H A D_tensor_grad.py365 def softmax_grad(orig, grad): function
376 return softmax_grad(sm, grad)
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/relay/op/
H A D_tensor_grad.py492 def softmax_grad(orig, grad): function
503 return softmax_grad(sm, grad)