| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 35 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest()
57 TensorShape x_shape({1, 2, 3}); in TEST_F() local
67 TensorShape x_shape({1, 2, 3}); in TEST_F() local
77 TensorShape x_shape({4, 2, 3}); in TEST_F() local
86 TensorShape x_shape({4, 2, 3}); in TEST_F() local
102 TensorShape x_shape({5, 2}); in TEST_F() local
112 TensorShape x_shape({}); in TEST_F() local
120 TensorShape x_shape({5, 2}); in TEST_F() local
128 TensorShape x_shape({5, 2, 5, 2}); in TEST_F() local
136 TensorShape x_shape({5, 2}); in TEST_F() local
[all …]
|
| D | nn_grad_test.cc | 54 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest()
211 TensorShape x_shape({5, 2}); in TEST_F() local
237 TensorShape x_shape({1, 2, 2, 1}); in TEST_F() local
250 TensorShape x_shape({1, 2, 2, 1}); in TEST_F() local
263 TensorShape x_shape({1, 3, 3, 3, 1}); in TEST_F() local
276 TensorShape x_shape({1, 2, 2, 1}); in TEST_F() local
287 TensorShape x_shape({1, 3, 3, 3, 1}); in TEST_F() local
298 TensorShape x_shape({1, 1, 2, 1}); in TEST_F() local
319 TensorShape x_shape({1, 3, 7, 1}); in TEST_F() local
331 TensorShape x_shape({1, 3, 7, 1}); in TEST_F() local
|
| D | image_grad_test.cc | 81 TensorShape x_shape({1, 2, 2, 1}); in TestResizedShapeForType() local
114 TensorShape x_shape({1, 4, 6, 1}); in TestResizeToSmallerAndAlign() local
129 TensorShape x_shape({1, 2, 3, 1}); in TestResizeToLargerAndAlign() local
211 void TestScaleAndTranslate(const TensorShape x_shape, const int out_height, in TestScaleAndTranslate()
326 TensorShape x_shape({1, 4, 2, 1}); in TestCropAndResize() local
|
| D | manip_grad_test.cc | 31 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest()
|
| D | math_grad_test.cc | 602 TensorShape x_shape = shapes[0]; in TestMatMulGrad() local
631 TensorShape x_shape; in RandMatMulShapes() local
757 TensorShape x_shape({2, 3, 5, 7}); in TEST_F() local
766 TensorShape x_shape({2, 3, 5, 7}); in TEST_F() local
775 TensorShape x_shape({2, 3}); in TEST_F() local
786 TensorShape x_shape({2, 3}); in TEST_F() local
805 TensorShape x_shape({1}); in TEST_F() local
815 TensorShape x_shape({1}); in TEST_F() local
861 TensorShape x_shape({3, 2, 5}); in TEST_F() local
870 TensorShape x_shape({3, 2, 5}); in TEST_F() local
[all …]
|
| D | image_grad.cc | 38 auto x_shape = Slice(scope, Shape(scope, op.input(0)), {1}, {2}); in ResizeNearestNeighborGradHelper() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker.py | 57 def _compute_theoretical_jacobian(x, x_shape, x_data, dy, dy_shape, dx, argument
135 def _compute_numeric_jacobian(x, x_shape, x_data, y, y_shape, delta, argument
212 x_shape, argument
246 x_shape, argument
278 x_shape, argument
355 x_shape, argument
|
| D | nn_fused_batchnorm_test.py | 61 x_shape, argument
142 x_shape, argument
185 def _compute_gradient_error_float16(self, x, x32, x_shape, y, y32, y_shape): argument
223 x_shape, argument
286 x_shape, argument
389 def _runtests(self, x_shape, is_training, gradient_test=False, argument
|
| D | image_ops_test.py | 1717 x_shape, argument
1736 x_shape, argument
2046 x_shape, argument
2065 x_shape, argument
3083 x_shape, argument
3100 x_shape, argument
3624 def _assertResizeEqual(self, x, x_shape, y, y_shape, argument
3637 def _assertResizeCheckShape(self, x, x_shape, target_shape, argument
3712 x_shape, argument
3728 x_shape, argument
[all …]
|
| D | nn_batchnorm_test.py | 303 def _testBatchNormArbitraryShapes(self, x_shape, param_shape, atol=0.0001, argument
373 def _testSuffStats(self, x_shape, axes, shift, keep_dims, has_shape): argument
|
| D | nn_test.py | 149 def testGradient(self, x_shape): argument
236 def testGradient(self, x_shape): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_op.cc | 37 const TensorShape x_shape = ctx->InputShape(0); in Compile() local
75 const TensorShape x_shape = ctx->InputShape(0); in Compile() local
|
| D | beta_op.cc | 39 const TensorShape& x_shape = ctx->InputShape(2); in Compile() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | inplace_ops.cc | 45 const TensorShape x_shape = ctx->InputShape(0); in Compile() local
105 const TensorShape x_shape = ctx->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_mul_op_test.cc | 38 void TestMul(const std::vector<int64>& x_shape, in TestMul()
87 void TestMulShape(const std::vector<int64>& x_shape, in TestMulShape()
137 void TimeMul(const std::vector<int64>& x_shape, in TimeMul()
|
| D | quantized_add_op_test.cc | 38 void TestAdd(const std::vector<int64>& x_shape, in TestAdd()
87 void TestAddShape(const std::vector<int64>& x_shape, in TestAddShape()
137 void TimeAdd(const std::vector<int64>& x_shape, in TimeAdd()
|
| D | betainc_op.cc | 49 const TensorShape& x_shape = x.shape(); in Compute() local
|
| D | fused_batch_norm_op.cc | 1268 auto x_shape = x.shape(); in ComputeWithReservedSpace() local
1429 auto x_shape = x.shape(); in ComputeWithReservedSpace() local
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradient_checker_test.cc | 126 TensorShape x_shape({4, 3}); in TEST() local
142 TensorShape x_shape({5, 2}); in TEST() local
157 TensorShape x_shape({1, 2, 3}); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander_test.cc | 84 auto x_shape = lx.shape(); in TEST_P() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | atrous_convolution_test.py | 254 def _test_gradient(self, x_shape, f_shape, dilation_rate, padding): argument
|
| D | sparse_tensor_dense_matmul_op_test.py | 274 def _sparse_tensor_dense_vs_dense_matmul_benchmark_sparse(x_ind, x_val, x_shape, argument
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | input_layer_test.py | 81 self, x_shape, x_dtype, y_shape, y_dtype, color='red', argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | type_spec_test.py | 59 def __init__(self, x_shape, x_dtype, y_shape, y_dtype, color="red"): argument
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 248 std::vector<int64> x_shape({10, 100}); in TestQuantizeMul() local
283 std::vector<int64> x_shape({10, 100}); in TestQuantizeAdd() local
|