| /external/tensorflow/tensorflow/python/tpu/tests/ |
| D | tpu_embedding_v1_correctness_test.py | 197 gradients, optimizer, argument
207 gradients, optimizer, argument
222 gradients, optimizer, argument
243 gradients, optimizer, argument
|
| D | tpu_embedding_base_test.py | 481 gradients, optimizer, argument
490 gradients, optimizer, argument
504 self, embedding_table_before, gradients, optimizer, variable): argument
534 gradients, optimizer, argument
555 gradients, optimizer, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 104 HloInstruction* gradients = in TEST_F() local
146 HloInstruction* gradients = in TEST_F() local
177 HloInstruction* gradients = in TEST_F() local
207 HloInstruction* gradients = in TEST_F() local
236 HloInstruction* gradients = in TEST_F() local
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | not_differentiable.cc | 18 namespace gradients { namespace
|
| D | array_grad.cc | 20 namespace gradients { namespace
|
| D | array_grad_test.cc | 28 namespace gradients { namespace
|
| D | custom_gradient_test.cc | 29 namespace gradients { namespace
|
| D | grad_test_helper.cc | 22 namespace gradients { namespace
|
| D | nn_grad.cc | 33 namespace gradients { namespace
|
| /external/tensorflow/tensorflow/c/experimental/gradients/tape/ |
| D | tape_context.cc | 21 namespace gradients { namespace
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_variants_test.cc | 1150 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1169 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1189 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1210 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1233 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1259 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1286 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1302 auto gradients = ConstantR3FromArray3D<float>( in XLA_TEST_F() local
1318 auto gradients = in XLA_TEST_F() local
1337 auto gradients = ConstantLiteral(&builder, gradients_literal); in XLA_TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | tape.cc | 28 namespace gradients { namespace
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | xent_op_deterministic_test.py | 125 def gradients(seed): function
|
| D | sparse_xent_op_deterministic_test.py | 127 def gradients(seed=789): function
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | xent_op_d9m_test.py | 124 def gradients(seed): function
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_xent_op_d9m_test.py | 123 def gradients(seed): function
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | model_test.cc | 787 Model::ParameterGradients gradients; in TEST() local
846 Model::ParameterGradients gradients; in TEST_P() local
894 Model::ParameterGradients gradients; in TEST() local
926 Model::ParameterGradients gradients; in TEST() local
958 Model::ParameterGradients gradients; in TEST() local
990 Model::ParameterGradients gradients; in TEST() local
|
| D | model.cc | 247 inline void UpdateParameterValues(const Node::ParameterGradients& gradients, in UpdateParameterValues()
423 ParameterGradients* gradients, NodeValues* output_times, in OutputTimeLocked()
605 ParameterGradients* gradients, NodeValues* output_times, in OutputTimeLocked()
753 ParameterGradients* gradients, NodeValues* output_times, in OutputTimeLocked()
888 ParameterGradients* gradients, NodeValues* output_times, in OutputTimeLocked()
1109 ParameterGradients* gradients, NodeValues* output_times, in OutputTimeLocked()
1200 ParameterGradients* gradients, NodeValues* output_times, in OutputTimeLocked()
2381 Model::ParameterGradients gradients; in OptimizeGradientDescent() local
2710 Model::ParameterGradients* gradients) { in OutputTime()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_impl.py | 40 def gradients(ys, function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | relu_op.cc | 86 auto gradients = ctx->Input("gradients"); in Compile() local
|
| D | lrn_ops.cc | 165 xla::XlaOp gradients = xla::Add( in Compile() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradients_test.cc | 42 namespace gradients { namespace
|
| D | gradient_checker_test.cc | 27 namespace gradients { namespace
|
| D | gradient_checker.cc | 25 namespace gradients { namespace
|
| /external/tensorflow/tensorflow/c/experimental/ops/ |
| D | nn_ops.cc | 66 Status ReluGrad(AbstractContext* ctx, AbstractTensorHandle* const gradients, in ReluGrad()
|