Searched refs:TapeContext (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/gradients/tape/ |
| D | tape_context.cc | 22 TapeContext::TapeContext(AbstractContext* c, Tape* tape, in TapeContext() function in tensorflow::gradients::TapeContext
28 void TapeContext::Release() { in Release()
32 TapeContext::~TapeContext() { in ~TapeContext()
36 TapeOperation* TapeContext::CreateOperation() { in CreateOperation()
39 Status TapeContext::RegisterFunction(AbstractFunction* f) { in RegisterFunction()
42 Status TapeContext::RemoveFunction(const string& func) { in RemoveFunction()
|
| D | tape_context.h | 23 class TapeContext : public AbstractContext {
25 explicit TapeContext(AbstractContext*, Tape*, const GradientRegistry&);
34 ~TapeContext() override;
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | tape.cc | 68 py::class_<TapeContext, AbstractContext>(m, "TapeContext") in PYBIND11_MODULE()
71 return new TapeContext(ctx, tape, *registry); in PYBIND11_MODULE()
|
| D | tape.py | 33 self._tape_context = _tape.TapeContext(
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | mnist_gradients_testutil.cc | 53 AbstractContextPtr tape_ctx(new TapeContext(ctx, tape, registry)); in AddGradModel()
77 AbstractContextPtr tape_ctx(new TapeContext(ctx, tape, registry)); in MatMulGradModel()
176 AbstractContextPtr tape_ctx(new TapeContext(ctx, tape, registry)); in MNISTGradModel()
|
| D | gradients_test.cc | 83 AbstractContextPtr tape_ctx(new TapeContext(ctx, tape.get(), registry)); in IdentityNGradModel()
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | nn_grad_test.cc | 47 AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, registry)); in ReluGradModel()
88 AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, registry)); in SparseSoftmaxCrossEntropyWithLogitsGradModel()
118 AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, registry)); in BiasAddGradModel()
|
| D | grad_test_helper.cc | 120 AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, grad_registry)); in BuildGradModel()
|