| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale.py | 85 def update(self, grads): argument
246 def update(self, grads): argument
257 def _is_all_finite(grads): argument
365 def update(self, grads): argument
371 def get_is_finite(grads): argument
|
| D | loss_scale_optimizer.py | 138 def _unscale_grads(self, grads): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cudnn_rnn_grad.py | 21 def _cudnn_rnn_backward(op, *grads):
73 def _cudnn_rnn_backwardv3(op, *grads):
|
| D | gradients_util.py | 227 def _VerifyGeneratedGradients(grads, op): argument
745 def _HasAnyNotNoneGrads(grads, op): argument
757 def _UpdatePendingAndEnqueueReady(grads, op, queue, pending_count, loop_state, argument
795 def _SetGrad(grads, t, grad): argument
819 def _GetGrad(grads, t, unconnected_gradients): argument
842 def _GetGrads(grads, op): argument
937 def _AggregatedGrads(grads, argument
|
| D | rnn_grad.py | 20 def _block_lstm_grad(op, *grads):
|
| D | cond_v2.py | 108 def _IfGrad(op, *grads): # pylint: disable=invalid-name
390 def _grad_fn(func_graph, grads): argument
428 def _create_grad_func(func_graph, grads, name): argument
1067 def _CaseGrad(op, *grads): # pylint: disable=invalid-name
|
| D | optional_grad.py | 28 def _OptionalGetValueGrad(unused_op, *grads):
|
| D | sparse_grad.py | 59 def _SparseAddGrad(op, *grads):
113 def _SparseSliceGrad(op, *grads):
|
| D | data_flow_grad.py | 27 def _DynamicPartitionGrads(op, *grads):
|
| D | while_v2_indexed_slices_rewriter.py | 28 def rewrite_grad_indexed_slices(grads, body_grad_graph, loop_vars, argument
|
| D | while_v2.py | 345 def _WhileGrad(op, *grads): # pylint: disable=invalid-name
655 def _create_grad_func(ys, xs, grads, cond_graph, body_graph, name, while_op, argument
807 def _get_structured_grad_output(outputs, grads, body_grad_graph): argument
|
| D | array_grad.py | 44 def _UnpackGrad(op, *grads):
352 def _SplitGrad(op, *grads):
357 def _SplitVGrad(op, *grads):
|
| /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/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 64 def _is_all_finite(grads): argument
212 def update(self, grads): argument
542 def get_unscaled_gradients(self, grads): argument
628 def _if_should_apply_grads(grads): argument
645 def _apply_gradients_cross_replica(distribution, grads, wrapped_vars, argument
665 def _apply_gradients(self, grads, wrapped_vars, name): argument
|
| /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/ops/ragged/ |
| D | ragged_conversion_ops.py | 145 def _ragged_tensor_from_variant_grad(op, *grads):
|
| D | ragged_gather_ops.py | 493 def _ragged_gather_grad(op, *grads):
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | crop_and_resize_op.cc | 366 const Tensor& grads = context->input(0); in ComputeAsync() local
432 const Tensor& grads = context->input(0); in ComputeAsync() local
588 const Tensor& grads = context->input(0); in ComputeAsync() local
648 const Tensor& grads = context->input(0); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_array_ops_decomposition.cc | 189 llvm::StringMap<Value> grads; member
526 const llvm::SmallDenseMap<int64_t, llvm::SmallVector<string, 4>>& grads, in ChangeFunctionInputSignature()
561 auto grads = AccessedGradients({body, cond}, module); in HandleWhileOp() local
646 auto grads = AccessedGradients({then_branch, else_branch}, module); in HandleIfOp() local
791 auto grads = AccessedGradients({lowered_callee}, module); in HandlePartitionedCallOp() local
|
| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | sparse_csr_matrix_grad.py | 55 def _CSRSparseMatrixToSparseTensorGrad(op, *grads):
|
| /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/python/eager/ |
| D | backprop.py | 605 def aggregate_indexed_slices_gradients(grads): argument
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function_test.cc | 1336 std::vector<std::pair<string, string>> grads = GetGradDefs(gdef); in TEST_F() local
1380 std::vector<std::pair<string, string>> grads = GetGradDefs(gdef); in TEST_F() local
1421 std::vector<std::pair<string, string>> grads = GetGradDefs(gdef); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 685 auto grads = xla::Add(grads_hi, grads_lo); // Want an unsigned add. in Compile() local
|