Searched defs:tape_ctx (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | mnist_gradients_testutil.cc | 53 AbstractContextPtr tape_ctx(new TapeContext(ctx, tape, registry)); in AddGradModel() local
77 AbstractContextPtr tape_ctx(new TapeContext(ctx, tape, registry)); in MatMulGradModel() local
176 AbstractContextPtr tape_ctx(new TapeContext(ctx, tape, registry)); in MNISTGradModel() local
|
| D | gradients_test.cc | 83 AbstractContextPtr tape_ctx(new TapeContext(ctx, tape.get(), registry)); in IdentityNGradModel() local
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | nn_grad_test.cc | 47 AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, registry)); in ReluGradModel() local
88 AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, registry)); in SparseSoftmaxCrossEntropyWithLogitsGradModel() local
118 AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, registry)); in BiasAddGradModel() local
|
| D | grad_test_helper.cc | 120 AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, grad_registry)); in BuildGradModel() local
|