| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradient_checker.cc | 105 const std::vector<TensorShape>& x_shapes, in ComputeTheoreticalJacobianTranspose() argument
110 size_t x_num = x_shapes.size(); in ComputeTheoreticalJacobianTranspose()
161 const int64 x_size = x_shapes[x_idx].num_elements(); in ComputeTheoreticalJacobianTranspose()
206 const std::vector<TensorShape>& x_shapes, in ComputeNumericJacobianTranspose() argument
213 size_t x_num = x_shapes.size(); in ComputeNumericJacobianTranspose()
223 const int64 x_size = x_shapes[x_idx].num_elements(); in ComputeNumericJacobianTranspose()
297 const std::vector<TensorShape>& x_shapes, in InitJacobians() argument
301 const size_t x_num = x_shapes.size(); in InitJacobians()
309 x_shapes[x_idx].num_elements() * JacobianStride<X_T>::value; in InitJacobians()
326 const std::vector<TensorShape>& x_shapes, in ComputeGradientErrorInternal() argument
[all …]
|
| D | gradient_checker.h | 50 const std::vector<TensorShape>& x_shapes,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | concat_op_test.py | 394 x_shapes = [[20, 7, 3], [10, 7, 3], [14, 7, 3]]
398 for x_shape in x_shapes
403 err = gradient_checker.compute_gradient_error(xs, x_shapes, output,
409 x_shapes = [[20, 7, 3], [20, 3, 3], [20, 1, 3]]
413 for x_shape in x_shapes
418 err = gradient_checker.compute_gradient_error(xs, x_shapes, output,
424 x_shapes = [[20, 7, 3], [10, 7, 3], [14, 7, 3]]
428 for x_shape in x_shapes
434 err = gradient_checker.compute_gradient_error(xs, x_shapes, output,
440 x_shapes = [[20, 7, 3], [20, 3, 3], [20, 1, 3]]
[all …]
|
| /external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
| D | distort_image_ops_test.py | 82 x_shapes = [
96 for x_shape in x_shapes:
142 x_shapes = [
156 for x_shape in x_shapes:
206 x_shapes = [
221 for x_shape in x_shapes:
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | data_flow_grad_test.cc | 37 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest() argument
42 scope_, xs, x_shapes, ys, y_shapes, &max_error))); in RunTest()
|
| D | nn_grad_test.cc | 70 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest() argument
75 scope_, xs, x_shapes, ys, y_shapes, &max_error))); in RunTest()
|
| D | array_grad_test.cc | 44 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest() argument
49 scope_, xs, x_shapes, ys, y_shapes, &max_error))); in RunTest()
|
| D | math_grad_test.cc | 713 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest() argument
718 scope_, xs, x_shapes, ys, y_shapes, &max_error))); in RunTest()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | image_ops_test.py | 156 x_shapes = [
163 for x_shape in x_shapes:
254 x_shapes = [
268 for x_shape in x_shapes:
366 x_shapes = [
381 for x_shape in x_shapes:
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_test.py | 429 x_shapes = [
443 for x_shape in x_shapes:
909 x_shapes = [
924 for x_shape in x_shapes:
1377 x_shapes = [
1384 for x_shape in x_shapes:
1692 x_shapes = [[13, 9, 3], [5, 13, 9, 3]]
1693 for x_shape in x_shapes:
1737 x_shapes = [[240, 320, 3], [5, 240, 320, 3]]
1740 for x_shape, y_shape in zip(x_shapes, expected_y_shapes):
[all …]
|