Searched refs:AbstractContextPtr (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | unified_api_test.cc | 59 AbstractContextPtr ctx; in TEST_P()
109 AbstractContextPtr ctx; in TEST_P()
144 AbstractContextPtr ctx(BuildFunction("test_fn")); in TEST_P()
168 AbstractContextPtr ctx(BuildFunction("test_fn")); in TEST_P()
|
| D | gradients_test.cc | 83 AbstractContextPtr tape_ctx(new TapeContext(ctx, tape.get(), registry)); in IdentityNGradModel()
110 AbstractContextPtr ctx; in TEST_P()
158 AbstractContextPtr ctx; in TEST_P()
223 AbstractContextPtr ctx; in TEST_P()
|
| D | mnist_gradients_test.cc | 69 AbstractContextPtr ctx; in TEST_P()
140 AbstractContextPtr ctx; in TEST_P()
216 AbstractContextPtr ctx; in TEST_P()
296 AbstractContextPtr ctx; in TEST_P()
354 AbstractContextPtr ctx; in TEST_P()
449 AbstractContextPtr ctx; in TEST_P()
508 AbstractContextPtr ctx; in TEST_P()
|
| 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 | abstract_context.h | 77 using AbstractContextPtr = variable
|
| D | gradient_checker_test.cc | 107 AbstractContextPtr ctx_;
|
| D | unified_api_testutil.cc | 64 AbstractContextPtr func_ctx(BuildFunction(fn_name)); in RunModel()
|
| D | gradients_util.cc | 250 AbstractContextPtr func_ctx(BuildFunction(fn_name)); in RunModel()
|
| /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()
149 AbstractContextPtr ctx_;
|
| D | custom_gradient_test.cc | 92 AbstractContextPtr ctx; in TEST_P()
|
| D | grad_test_helper.cc | 120 AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, grad_registry)); in BuildGradModel()
|
| D | math_grad_test.cc | 105 AbstractContextPtr immediate_execution_ctx_;
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api.cc | 46 using tensorflow::AbstractContextPtr;
114 py::class_<AbstractContext, AbstractContextPtr>(m, "AbstractContext") in PYBIND11_MODULE()
|