| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 103 HloInstruction* gradients = in TEST_F() local
145 HloInstruction* gradients = in TEST_F() local
176 HloInstruction* gradients = in TEST_F() local
206 HloInstruction* gradients = in TEST_F() local
235 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 | 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
|
| D | nn_grad_test.cc | 26 namespace gradients { namespace
|
| /external/tensorflow/tensorflow/c/experimental/gradients/tape/ |
| D | tape_context.cc | 21 namespace gradients { namespace
|
| D | tape_operation.cc | 21 namespace gradients { namespace
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2_correctness_test.py | 391 gradients, argument
401 gradients, argument
416 gradients, argument
|
| D | tpu_embedding_v2.py | 519 def _compute_per_table_gradients( argument
577 def apply_gradients(self, gradients, name: Text = None): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | model_test.cc | 642 absl::flat_hash_map<string, double> gradients; in TEST() local
696 absl::flat_hash_map<string, double> gradients; in TEST_P() local
742 absl::flat_hash_map<string, double> gradients; in TEST() local
773 absl::flat_hash_map<string, double> gradients; in TEST() local
804 absl::flat_hash_map<string, double> gradients; in TEST() local
835 absl::flat_hash_map<string, double> gradients; in TEST() local
|
| D | model.cc | 81 const absl::flat_hash_map<string, double>& gradients, in UpdateParameterValues()
174 absl::flat_hash_map<string, double>* gradients, in OutputTimeLocked()
311 absl::flat_hash_map<string, double>* gradients, in OutputTimeLocked()
462 absl::flat_hash_map<string, double>* gradients, in OutputTimeLocked()
585 absl::flat_hash_map<string, double>* gradients, in OutputTimeLocked()
795 absl::flat_hash_map<string, double>* gradients, in OutputTimeLocked()
889 absl::flat_hash_map<string, double>* gradients, in OutputTimeLocked()
1818 absl::flat_hash_map<string, double> gradients; in OptimizeGradientDescent() local
1905 absl::flat_hash_map<string, double>* gradients) { in OutputTime()
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | tape.cc | 28 namespace gradients { namespace
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_variants_test.cc | 1151 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1170 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1190 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1211 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1234 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1260 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1287 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1303 auto gradients = ConstantR3FromArray3D<float>( in XLA_TEST_F() local
1319 auto gradients = in XLA_TEST_F() local
1339 auto gradients = ConstantLiteral(&builder, gradients_literal); in XLA_TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_impl.py | 44 def gradients(ys, function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | lrn_ops.cc | 165 xla::XlaOp gradients = xla::Add( in Compile() local
|
| D | relu_op.cc | 116 auto gradients = ctx->Input("gradients"); in Compile() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradient_checker.cc | 25 namespace gradients { namespace
|
| D | gradient_checker_test.cc | 27 namespace gradients { namespace
|
| D | mnist_gradients_testutil.cc | 35 namespace gradients { namespace
|
| D | gradients_test.cc | 42 namespace gradients { namespace
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 1259 const auto gradients = gradients_t->matrix<float>(); in Compute() local
1327 const auto gradients = gradients_t->matrix<float>(); in Compute() local
1469 const TTypes<float>::ConstMatrix& gradients, in AddInstanceStatsToMap()
1493 const TTypes<float>::ConstMatrix& gradients, in AddRangeStats()
1532 const auto gradients = gradients_t->matrix<float>(); in Compute() local
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | resnet50_test.py | 54 def apply_gradients(model, optimizer, gradients): argument
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_loss_calculator.h | 118 VectorOut* loss, std::vector<MatrixOut>* gradients, in CalculateLoss()
|