| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | custom_training_loop_optimizer_test.py | 63 def step_fn(grads): argument
90 def step_fn(grads): argument
112 def step_fn(grads): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cudnn_rnn_grad.py | 25 def _cudnn_rnn_backward(op, *grads):
77 def _cudnn_rnn_backwardv3(op, *grads):
|
| D | gradients_util.py | 238 def _VerifyGeneratedGradients(grads, op): argument
734 def _HasAnyNotNoneGrads(grads, op): argument
746 def _UpdatePendingAndEnqueueReady(grads, op, queue, pending_count, loop_state, argument
784 def _SetGrad(grads, t, grad): argument
808 def _GetGrad(grads, t, unconnected_gradients): argument
831 def _GetGrads(grads, op): argument
926 def _AggregatedGrads(grads, argument
|
| D | rnn_grad.py | 24 def _block_lstm_grad(op, *grads):
|
| D | cond_v2.py | 111 def _IfGrad(op, *grads): # pylint: disable=invalid-name
353 def _grad_fn(func_graph, grads): argument
391 def _create_grad_func(func_graph, grads, name): argument
1026 def _CaseGrad(op, *grads): # pylint: disable=invalid-name
|
| D | optional_grad.py | 32 def _OptionalGetValueGrad(unused_op, *grads):
|
| D | data_flow_grad.py | 31 def _DynamicPartitionGrads(op, *grads):
|
| D | sparse_grad.py | 66 def _SparseAddGrad(op, *grads):
120 def _SparseSliceGrad(op, *grads):
|
| D | while_v2_indexed_slices_rewriter.py | 31 def rewrite_grad_indexed_slices(grads, body_grad_graph, loop_vars, argument
|
| D | while_v2.py | 311 def _WhileGrad(op, *grads): # pylint: disable=invalid-name
608 def _create_grad_func(ys, xs, grads, cond_graph, body_graph, name, while_op, argument
758 def _get_structured_grad_output(outputs, grads, body_grad_graph): argument
|
| D | array_grad.py | 48 def _UnpackGrad(op, *grads):
342 def _SplitGrad(op, *grads):
347 def _SplitVGrad(op, *grads):
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale.py | 102 def update(self, grads): argument
270 def update(self, grads): argument
281 def _is_all_finite(grads): argument
399 def update(self, grads): argument
405 def get_is_finite(grads): argument
|
| D | loss_scaling_gradient_tape.py | 256 def cond(grads, ready_to_update, is_first_iteration): argument
270 def body(grads, ready_to_update, is_first_iteration): argument
|
| D | loss_scale_optimizer.py | 142 def _unscale_grads(self, grads): argument
|
| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/op/core/ |
| D | GradientsTest.java | 47 Gradients grads = Gradients.create(scope, y1, Arrays.asList(x, y0)); in createGradients() local
74 Gradients grads = Gradients.create(scope, Arrays.asList(y0, y1), Arrays.asList(x)); in createGradientsWithSum() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gradients.cc | 102 static Node* AddSymGrad(Graph* g, Node* n, gtl::ArraySlice<NodeOut> grads) { in AddSymGrad()
234 auto* grads = &iter->second; in BackpropAlongEdge() local
307 const auto& grads = iter->second; in SumGradients() local
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 183 def _is_all_finite(grads): argument
328 def update(self, grads): argument
346 def get_is_finite(grads): argument
650 def get_unscaled_gradients(self, grads): argument
748 def _apply_gradients(self, grads, wrapped_vars, name, argument
|
| D | loss_scale_optimizer_test.py | 592 def update(self, grads): argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_conversion_ops.py | 149 def _ragged_tensor_from_variant_grad(op, *grads):
|
| D | ragged_gather_ops.py | 473 def _ragged_gather_grad(op, *grads):
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | crop_and_resize_op.cc | 364 const Tensor& grads = context->input(0); in ComputeAsync() local
420 const Tensor& grads = context->input(0); in ComputeAsync() local
570 const Tensor& grads = context->input(0); in ComputeAsync() local
621 const Tensor& grads = context->input(0); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_array_ops_decomposition.cc | 190 llvm::StringMap<Value> grads; member
527 const llvm::SmallDenseMap<int64_t, llvm::SmallVector<string, 4>>& grads, in ChangeFunctionInputSignature()
562 auto grads = AccessedGradients({body, cond}, module); in HandleWhileOp() local
646 auto grads = AccessedGradients({then_branch, else_branch}, module); in HandleIfOp() local
790 auto grads = AccessedGradients({lowered_callee}, module); in HandlePartitionedCallOp() local
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradients.cc | 159 auto* grads = &iter->second; in BackpropAlongEdge() local
349 const auto& grads = iter->second; in SumGradients() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lrn_op_test.cc | 205 Tensor grads(DT_FLOAT, TensorShape({batches, rows, cols, depth})); in MakeRNGrad() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradients_util.cc | 174 Status UpdateWeights(AbstractContext* ctx, vector<AbstractTensorHandle*>& grads, in UpdateWeights()
|