Searched refs:BuildGradModel (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/ |
| H A D | math_grad_test.cc | 139 AddModel, BuildGradModel(AddModel, registry_), in TEST_P()
157 ExpModel, BuildGradModel(ExpModel, registry_), in TEST_P()
203 MatMulModel, BuildGradModel(MatMulModel, registry_), in TEST_P()
257 Model MatMulGradModel = BuildGradModel(MatMulModel, registry_); in TEST_P()
288 SqrtModel, BuildGradModel(SqrtModel, registry_), in TEST_P()
306 NegModel, BuildGradModel(NegModel, registry_), in TEST_P()
333 SubModel, BuildGradModel(SubModel, registry_), in TEST_P()
360 MulModel, BuildGradModel(MulModel, registry_), in TEST_P()
378 Log1pModel, BuildGradModel(Log1pModel, registry_), in TEST_P()
386 auto DivNoNanGradModel = BuildGradModel(DivNoNanModel, registry_); in TEST_P()
|
| H A D | nn_grad_test.cc | 99 auto ReluGradModel = BuildGradModel(ReluModel, registry_); in TEST_P()
172 BuildGradModel(SparseSoftmaxCrossEntropyWithLogitsModel, registry_), in TEST_P()
208 BiasAddModel, BuildGradModel(BiasAddModel, registry_), in TEST_P()
|
| H A D | grad_test_helper.h | 33 Model BuildGradModel(Model forward, GradientRegistry registry);
|
| H A D | array_grad_test.cc | 104 auto IdentityNGradModel = BuildGradModel(IdentityNModel, registry_); in TEST_P()
|
| H A D | grad_test_helper.cc | 109 Model BuildGradModel(Model forward, GradientRegistry registry) { in BuildGradModel() function
|