Searched refs:x_raw (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | math_grad_test.cc | 117 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
119 TestScalarTensorHandle(immediate_execution_ctx_.get(), 2.0f, &x_raw); in TEST_P()
121 x.reset(x_raw); in TEST_P()
144 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
146 TestScalarTensorHandle(immediate_execution_ctx_.get(), 2.0f, &x_raw); in TEST_P()
148 x.reset(x_raw); in TEST_P()
210 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
212 TestScalarTensorHandle(immediate_execution_ctx_.get(), 2.0f, &x_raw); in TEST_P()
214 x.reset(x_raw); in TEST_P()
228 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
[all …]
|
| D | custom_gradient_test.cc | 103 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
104 Status s = TestScalarTensorHandle(ctx.get(), 1.0f, &x_raw); in TEST_P()
106 x.reset(x_raw); in TEST_P()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | unified_api_test.cc | 70 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
71 Status s = TestScalarTensorHandle(ctx.get(), 2.0f, &x_raw); in TEST_P()
73 x.reset(x_raw); in TEST_P()
120 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
124 TestTensorHandleWithDimsFloat(ctx.get(), data, dim_sizes, 2, &x_raw); in TEST_P()
126 x.reset(x_raw); in TEST_P()
147 tracing::TracingTensorHandle* x_raw = nullptr; in TEST_P() local
150 DT_FLOAT, shape, &x_raw); in TEST_P()
152 x.reset(x_raw); in TEST_P()
171 tracing::TracingTensorHandle* x_raw = nullptr; in TEST_P() local
[all …]
|
| D | gradients_test.cc | 121 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
122 Status s = TestScalarTensorHandle(ctx.get(), 1.0f, &x_raw); in TEST_P()
124 x1.reset(x_raw); in TEST_P()
128 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
129 Status s = TestScalarTensorHandle(ctx.get(), 1.0f, &x_raw); in TEST_P()
131 x2.reset(x_raw); in TEST_P()
169 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
170 Status s = TestScalarTensorHandle(ctx.get(), 1.0f, &x_raw); in TEST_P()
172 t.reset(x_raw); in TEST_P()
234 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
[all …]
|
| D | gradient_checker_test.cc | 145 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
146 Status s = TestScalarTensorHandle(ctx_.get(), 2.0f, &x_raw); in TEST_P()
148 x.reset(x_raw); in TEST_P()
|
| D | mnist_gradients_test.cc | 460 AbstractTensorHandle* x_raw = nullptr; in TEST_P() local
461 Status s = ScalarTensorHandle(ctx.get(), 1.5f, &x_raw); in TEST_P()
463 eta.reset(x_raw); in TEST_P()
|