Searched refs:RegisterGradient (Results 1 – 25 of 52) sorted by relevance
123
79 @ops.RegisterGradient("TensorArrayRead")80 @ops.RegisterGradient("TensorArrayReadV2")81 @ops.RegisterGradient("TensorArrayReadV3")110 @ops.RegisterGradient("TensorArrayWrite")111 @ops.RegisterGradient("TensorArrayWriteV2")112 @ops.RegisterGradient("TensorArrayWriteV3")136 @ops.RegisterGradient("TensorArrayGather")137 @ops.RegisterGradient("TensorArrayGatherV2")138 @ops.RegisterGradient("TensorArrayGatherV3")167 @ops.RegisterGradient("TensorArrayScatter")[all …]
38 @ops.RegisterGradient("ArgMax")44 @ops.RegisterGradient("ArgMin")54 @ops.RegisterGradient("Sum")109 @ops.RegisterGradient("Max")115 @ops.RegisterGradient("Min")120 @ops.RegisterGradient("Mean")140 @ops.RegisterGradient("Prod")188 @ops.RegisterGradient("SegmentSum")194 @ops.RegisterGradient("SegmentMean")207 @ops.RegisterGradient("SparseSegmentSum")[all …]
35 @ops.RegisterGradient("Pack")41 @ops.RegisterGradient("Unpack")210 @ops.RegisterGradient("Concat")220 @ops.RegisterGradient("ConcatV2")229 @ops.RegisterGradient("Slice")254 @ops.RegisterGradient("StridedSlice")280 @ops.RegisterGradient("StridedSliceGrad")299 @ops.RegisterGradient("Split")304 @ops.RegisterGradient("SplitV")317 @ops.RegisterGradient("Diag")[all …]
32 @ops.RegisterGradient("Conv2DBackpropInput")65 @ops.RegisterGradient("Conv2DBackpropFilter")88 @ops.RegisterGradient("DepthwiseConv2dNativeBackpropInput")119 @ops.RegisterGradient("DepthwiseConv2dNativeBackpropFilter")140 @ops.RegisterGradient("Conv3D")163 @ops.RegisterGradient("Conv3DBackpropInputV2")186 @ops.RegisterGradient("Conv3DBackpropFilterV2")208 @ops.RegisterGradient("AvgPool3D")219 @ops.RegisterGradient("AvgPool3DGrad")230 @ops.RegisterGradient("MaxPool3D")[all …]
36 @ops.RegisterGradient("SparseReorder")65 @ops.RegisterGradient("SparseAdd")99 @ops.RegisterGradient("SparseTensorDenseAdd")106 @ops.RegisterGradient("SparseReduceSum")119 @ops.RegisterGradient("SparseSlice")148 @ops.RegisterGradient("SparseTensorDenseMatMul")204 @ops.RegisterGradient("SparseDenseCwiseAdd")244 @ops.RegisterGradient("SparseDenseCwiseMul")250 @ops.RegisterGradient("SparseDenseCwiseDiv")256 @ops.RegisterGradient("SparseSoftmax")[all …]
91 ops.RegisterGradient("Switch")(_SwitchGrad)92 ops.RegisterGradient("RefSwitch")(_SwitchGrad)95 @ops.RegisterGradient("Merge")141 @ops.RegisterGradient("RefMerge")146 @ops.RegisterGradient("Exit")185 ops.RegisterGradient("RefExit")(_ExitGrad)188 @ops.RegisterGradient("NextIteration")197 @ops.RegisterGradient("RefNextIteration")202 @ops.RegisterGradient("Enter")235 @ops.RegisterGradient("RefEnter")[all …]
174 @ops.RegisterGradient("TensorListPushBack")182 @ops.RegisterGradient("TensorListPopBack")192 @ops.RegisterGradient("TensorListStack")197 @ops.RegisterGradient("TensorListConcat")198 @ops.RegisterGradient("TensorListConcatV2")212 @ops.RegisterGradient("TensorListSplit")224 @ops.RegisterGradient("TensorListFromTensor")246 @ops.RegisterGradient("TensorListGetItem")262 @ops.RegisterGradient("TensorListSetItem")277 @ops.RegisterGradient("TensorListResize")[all …]
27 @ops.RegisterGradient("ResizeNearestNeighbor")52 @ops.RegisterGradient("ResizeBilinear")71 @ops.RegisterGradient("ScaleAndTranslate")93 @ops.RegisterGradient("ResizeBicubic")115 @ops.RegisterGradient("CropAndResize")
38 @ops.RegisterGradient("MatrixInverse")46 @ops.RegisterGradient("MatrixDeterminant")58 @ops.RegisterGradient("MatrixSquareRoot")123 @ops.RegisterGradient("LogMatrixDeterminant")134 @ops.RegisterGradient("Cholesky")160 @ops.RegisterGradient("Qr")190 @ops.RegisterGradient("MatrixSolve")204 @ops.RegisterGradient("MatrixSolveLs")283 @ops.RegisterGradient("MatrixTriangularSolve")303 @ops.RegisterGradient("SelfAdjointEigV2")[all …]
30 @ops.RegisterGradient("DynamicPartition")47 @ops.RegisterGradient("DynamicStitch")48 @ops.RegisterGradient("ParallelDynamicStitch")
24 @ops.RegisterGradient("CudnnRNN")50 @ops.RegisterGradient("CudnnRNNV2")76 @ops.RegisterGradient("CudnnRNNV3")
25 @ops.RegisterGradient("OptionalFromValue")31 @ops.RegisterGradient("OptionalGetValue")
50 @ops.RegisterGradient('NcclAllReduce')149 @ops.RegisterGradient('NcclReduce')189 @ops.RegisterGradient('NcclBroadcast')
25 @ops.RegisterGradient("Roll")
175 @_ops.RegisterGradient("FFT")181 @_ops.RegisterGradient("IFFT")189 @_ops.RegisterGradient("FFT2D")195 @_ops.RegisterGradient("IFFT2D")203 @_ops.RegisterGradient("FFT3D")209 @_ops.RegisterGradient("IFFT3D")328 _ops.RegisterGradient("RFFT")(_rfft_grad_helper(1, irfft))329 _ops.RegisterGradient("IRFFT")(_irfft_grad_helper(1, rfft))330 _ops.RegisterGradient("RFFT2D")(_rfft_grad_helper(2, irfft2d))331 _ops.RegisterGradient("IRFFT2D")(_irfft_grad_helper(2, rfft2d))
77 @ops.RegisterGradient("AllToAll")137 @ops.RegisterGradient("CollectivePermute")146 @ops.RegisterGradient("CrossReplicaSum")163 @ops.RegisterGradient("TPUEmbeddingActivations")
1 path: "tensorflow.RegisterGradient"3 is_instance: "<class \'tensorflow.python.framework.ops.RegisterGradient\'>"
45 @ops.RegisterGradient('RoutingFunction')122 @ops.RegisterGradient('StochasticHardRoutingFunction')201 @ops.RegisterGradient('KFeatureRoutingFunction')
31 @ops.RegisterGradient("Batch")46 @ops.RegisterGradient("Unbatch")
24 @ops.RegisterGradient("XlaClusterOutput")
356 @ops.RegisterGradient("DebugGradientIdentity")366 @ops.RegisterGradient("DebugGradientRefIdentity")
27 @ops.RegisterGradient("ZeroOut")
34 @ops.RegisterGradient("PeriodicResample")
64 @ops.RegisterGradient("Resampler")