MCPcopy Create free account
hub / github.com/alibaba/nann / _TopKGrad

Function _TopKGrad

tensorflow/tensorflow/python/ops/nn_grad.py:1073–1114  ·  view source on GitHub ↗

Return the gradients for TopK. Args: op: The TopKOp for which we need to generate gradients. grad: Tensor. The gradients passed to the TopKOp. Returns: A list of two tensors, the first being the gradient w.r.t to the input and TopK, and the second being the gradient w.r.t. to t

(op, grad, _)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 9

reshapeMethod · 0.80
shapeMethod · 0.45
gatherMethod · 0.45
castMethod · 0.45
sizeMethod · 0.45
stackMethod · 0.45
expand_dimsMethod · 0.45
rangeMethod · 0.45
scatter_ndMethod · 0.45

Tested by

no test coverage detected