| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 37 def _ArgMaxGrad(op, grad): argument
43 def _ArgMinGrad(op, grad): argument
49 def _EuclideanNormGrad(op, grad): argument
63 def SmartBroadcastGradientArgs(x, y, grad): argument
142 def _SumGrad(op, grad): argument
213 def _MinOrMaxGrad(op, grad): argument
235 def _MaxGrad(op, grad): argument
241 def _MinGrad(op, grad): argument
246 def _MeanGrad(op, grad): argument
266 def _ProdGrad(op, grad): argument
[all …]
|
| D | array_grad.py | 38 def _PackGrad(op, grad): argument
49 def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index): argument
216 def _ConcatGrad(op, grad): argument
226 def _ConcatGradV2(op, grad): argument
235 def _SliceGrad(op, grad): argument
263 def _StridedSliceGrad(op, grad): argument
298 def _StridedSliceGradGrad(op, grad): argument
317 def _TensorStridedSliceUpdateGrad(op, grad): # pylint:disable=missing-function-docstring argument
370 def _DiagGrad(_, grad): argument
375 def _DiagPartGrad(_, grad): argument
[all …]
|
| D | nn_grad.py | 31 def _Conv2DBackpropInputGrad(op, grad): argument
68 def _Conv2DBackpropFilterGrad(op, grad): argument
95 def _DepthwiseConv2dNativeBackpropInputGrad(op, grad): argument
128 def _DepthwiseConv2dNativeBackpropFilterGrad(op, grad): argument
151 def _Conv3DGrad(op, grad): argument
174 def _Conv3DBackpropInputGrad(op, grad): argument
197 def _Conv3DBackpropFilterGrad(op, grad): argument
219 def _AvgPool3DGrad(op, grad): argument
230 def _AvgPool3DGradGrad(op, grad): argument
241 def _MaxPool3DGrad(op, grad): argument
[all …]
|
| D | control_flow_grad.py | 31 def _SwitchGrad(op, *grad):
92 def _MergeGrad(op, grad, _): argument
140 def _RefMergeGrad(op, grad, _): argument
145 def _ExitGrad(op, grad): argument
189 def _NextIterationGrad(_, grad): argument
198 def _RefNextIterationGrad(_, grad): argument
203 def _EnterGrad(op, grad): argument
239 def _RefEnterGrad(op, grad): argument
|
| D | image_grad.py | 25 def _ResizeNearestNeighborGrad(op, grad): argument
50 def _ResizeBilinearGrad(op, grad): argument
69 def _ScaleAndTranslateGrad(op, grad): argument
91 def _ResizeBicubicGrad(op, grad): argument
113 def _CropAndResizeGrad(op, grad): argument
167 def _RGBToHSVGrad(op, grad): argument
|
| D | linalg_grad.py | 47 def _MatrixInverseGrad(op, grad): argument
59 def _EinsumGrad(op, grad): argument
374 def _MatrixDeterminantGrad(op, grad): argument
386 def _MatrixSquareRootGrad(op, grad): argument
462 def _CholeskyGrad(op, grad): argument
556 def _MatrixSolveGrad(op, grad): argument
570 def _MatrixSolveLsGrad(op, grad): argument
579 def _Overdetermined(op, grad): argument
604 def _Underdetermined(op, grad): argument
649 def _BandedTriangularSolveGrad(op, grad): argument
[all …]
|
| D | sparse_grad.py | 142 def _SparseTensorDenseMatMulGrad(op, grad): argument
237 def _SparseDenseCwiseMulOrDivGrad(op, grad, is_mul): argument
272 def _SparseDenseCwiseMulGrad(op, grad): argument
278 def _SparseDenseCwiseDivGrad(op, grad): argument
284 def _SparseSoftmaxGrad(op, grad): argument
344 def _SparseToDenseGrad(op, grad): argument
|
| D | random_grad.py | 37 def _RandomGammaGrad(op, grad): # pylint: disable=invalid-name argument
77 def _StatelessRandomGammaV2Grad(op, grad): # pylint: disable=invalid-name argument
135 def _StatelessParameterizedTruncatedNormalGrad(op, grad): # pylint: disable=invalid-name argument
|
| D | tensor_array_grad.py | 83 def _TensorArrayReadGrad(op, grad): argument
146 def _TensorArrayGatherGrad(op, grad): argument
207 def _TensorArrayConcatGrad(op, grad, unused_lengths_grad): argument
|
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_grad.py | 21 def _RiscAbsGrad(_, grad): argument
28 def _RiscAddGrad(_, grad): argument
35 def _RiscBinaryArithmeticGrad(_, grad): argument
42 def _RiscBinaryComparisonGrad(_, grad): argument
49 def _RiscBitcastGrad(_, grad): argument
56 def _RiscBroadcastGrad(_, grad): argument
63 def _RiscCastGrad(_, grad): argument
70 def _RiscCholeskyGrad(_, grad): argument
77 def _RiscCeilGrad(_, grad): argument
84 def _RiscConcatGrad(_, grad): argument
[all …]
|
| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | sparse_csr_matrix_grad.py | 26 def _DenseToCSRSparseMatrixGrad(op, grad): argument
36 def _CSRSparseMatrixToDenseGrad(op, grad): argument
47 def _SparseTensorToCSRSparseMatrixGrad(op, grad): argument
111 def _SparseMatrixAddGrad(op, grad): argument
193 def _SparseMatrixTransposeGrad(op, grad): argument
208 def _SparseMatrixMatMulGrad(op, grad): argument
300 def _SparseMatrixSparseMatMulGrad(op, grad): argument
356 def _SparseMatrixMulGrad(op, grad): argument
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | fft_ops.py | 204 def _fft_size_for_grad(grad, rank): argument
209 def _fft_grad(_, grad): argument
215 def _ifft_grad(_, grad): argument
223 def _fft2d_grad(_, grad): argument
229 def _ifft2d_grad(_, grad): argument
237 def _fft3d_grad(_, grad): argument
243 def _ifft3d_grad(_, grad): argument
255 def _grad(op, grad): argument
337 def _grad(op, grad): argument
|
| /external/pytorch/test/dynamo/ |
| D | test_hooks.py | 21 def global_hook_0(grad): argument
25 def global_hook_1(grad): argument
29 def global_hook_2(grad): argument
160 def local_hook(grad): argument
177 def local_hook(grad): argument
194 def local_hook(grad): argument
210 def local_hook(grad): argument
213 def local_hook2(grad): argument
326 def my_hook(grad, *, k=0): argument
364 def my_hook(grad, *, k=0): argument
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | gradient_descent.py | 52 def _apply_dense(self, grad, var): argument
59 def _resource_apply_dense(self, grad, handle): argument
65 def _resource_apply_sparse_duplicate_indices(self, grad, handle, indices): argument
72 def _apply_sparse_duplicate_indices(self, grad, var): argument
|
| D | proximal_gradient_descent.py | 60 def _apply_dense(self, grad, var): argument
69 def _resource_apply_dense(self, grad, var): argument
78 def _apply_sparse(self, grad, var): argument
88 def _resource_apply_sparse(self, grad, var, indices): argument
|
| D | adadelta.py | 146 def _apply_dense(self, grad, var): argument
159 def _resource_apply_dense(self, grad, var): argument
172 def _apply_sparse(self, grad, var): argument
186 def _resource_apply_sparse(self, grad, var, indices): argument
|
| D | momentum.py | 165 def _apply_dense(self, grad, var): argument
175 def _resource_apply_dense(self, grad, var): argument
185 def _apply_sparse(self, grad, var): argument
195 def _resource_apply_sparse(self, grad, var, indices): argument
|
| D | proximal_adagrad.py | 90 def _apply_dense(self, grad, var): argument
98 def _resource_apply_dense(self, grad, var): argument
106 def _apply_sparse(self, grad, var): argument
115 def _resource_apply_sparse(self, grad, var, indices): argument
|
| D | adam.py | 213 def _apply_dense(self, grad, var): argument
230 def _resource_apply_dense(self, grad, var): argument
247 def _apply_sparse_shared(self, grad, var, indices, scatter_add): argument
273 def _apply_sparse(self, grad, var): argument
289 def _resource_apply_sparse(self, grad, var, indices): argument
|
| D | adagrad_da.py | 107 def _apply_dense(self, grad, var): argument
123 def _resource_apply_dense(self, grad, var): argument
139 def _apply_sparse(self, grad, var): argument
156 def _resource_apply_sparse(self, grad, var, indices): argument
|
| D | rmsprop.py | 209 def _apply_dense(self, grad, var): argument
237 def _resource_apply_dense(self, grad, var): argument
265 def _apply_sparse(self, grad, var): argument
295 def _resource_apply_sparse(self, grad, var, indices): argument
|
| D | adagrad.py | 159 def _apply_dense(self, grad, var): argument
168 def _resource_apply_dense(self, grad, var): argument
177 def _apply_sparse(self, grad, var): argument
187 def _resource_apply_sparse(self, grad, var, indices): argument
|
| D | ftrl.py | 163 def _apply_dense(self, grad, var): argument
195 def _resource_apply_dense(self, grad, var): argument
227 def _apply_sparse(self, grad, var): argument
261 def _resource_apply_sparse(self, grad, var, indices): argument
|
| D | training_ops_test.py | 69 def _testTypesForAdagrad(self, x, y, lr, grad, use_gpu=None): argument
89 grad, argument
134 grad, argument
222 def _testTypesForSparseAdagrad(self, x, y, lr, grad, indices, use_gpu): argument
248 grad, argument
288 grad, argument
413 def _testTypesForAdam(self, var, m, v, grad, use_gpu): argument
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 146 def _scale_grad(self, grad, loss_scale_reciprocal): argument
232 def _apply_sparse(self, grad, var): argument
236 def _apply_dense(self, grad, var): argument
240 def _resource_apply_sparse(self, grad, handle, indices): argument
244 def _resource_apply_dense(self, grad, handle): argument
|