| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A 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({2, 6}); in TEST_F() local
120 TensorShape x_shape({}); in TEST_F() local
128 TensorShape x_shape({5, 2}); in TEST_F() local
136 TensorShape x_shape({5, 2, 5, 2}); in TEST_F() local
[all …]
|
| H A D | linalg_grad_test.cc | 35 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest()
57 TensorShape x_shape({2, 3}); in TEST_F() local
65 TensorShape x_shape({3, 2, 3}); in TEST_F() local
73 TensorShape x_shape({2, 3}); in TEST_F() local
83 TensorShape x_shape({2, 3}); in TEST_F() local
97 TensorShape x_shape({3, 2, 3}); in TEST_F() local
107 TensorShape x_shape({3, 3}); in TEST_F() local
117 TensorShape x_shape({4, 3, 3}); in TEST_F() local
127 TensorShape x_shape({3}); in TEST_F() local
137 TensorShape x_shape({3}); in TEST_F() local
[all …]
|
| H A D | nn_grad_test.cc | 59 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest()
223 TensorShape x_shape({5, 2}); in TEST_F() local
249 TensorShape x_shape({1, 2, 2, 1}); in TEST_F() local
262 TensorShape x_shape({1, 2, 2, 1}); in TEST_F() local
275 TensorShape x_shape({1, 3, 3, 3, 1}); in TEST_F() local
288 TensorShape x_shape({1, 2, 2, 1}); in TEST_F() local
299 TensorShape x_shape({1, 3, 3, 3, 1}); in TEST_F() local
310 TensorShape x_shape({1, 1, 2, 1}); in TEST_F() local
331 TensorShape x_shape({1, 3, 7, 1}); in TEST_F() local
343 TensorShape x_shape({1, 3, 7, 1}); in TEST_F() local
|
| H A D | math_grad_test.cc | 611 TensorShape x_shape = shapes[0]; in TestMatMulGrad() local
640 TensorShape x_shape; in RandMatMulShapes() local
766 TensorShape x_shape({2, 3, 5, 7}); in TEST_F() local
775 TensorShape x_shape({2, 3, 5, 7}); in TEST_F() local
784 TensorShape x_shape({2, 3}); in TEST_F() local
795 TensorShape x_shape({2, 3}); in TEST_F() local
814 TensorShape x_shape({1}); in TEST_F() local
824 TensorShape x_shape({1}); in TEST_F() local
879 TensorShape x_shape({3, 2, 5}); in TEST_F() local
888 TensorShape x_shape({3, 2, 5}); in TEST_F() local
[all …]
|
| H A 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
|
| H A D | manip_grad_test.cc | 31 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest()
|
| H A D | functional_grad_test.cc | 36 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest()
|
| H A D | image_grad.cc | 44 auto x_shape = Slice(scope, Shape(scope, op.input(0)), {1}, {2}); in ResizeNearestNeighborGradHelper() local
|
| H A D | linalg_grad.cc | 432 const auto x_shape = Shape(scope, x); in EinsumGrad() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | gradient_checker.py | 54 def _compute_theoretical_jacobian(x, x_shape, x_data, dy, dy_shape, dx, argument
132 def _compute_numeric_jacobian(x, x_shape, x_data, y, y_shape, delta, argument
209 x_shape, argument
243 x_shape, argument
275 x_shape, argument
352 x_shape, argument
|
| H A 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
|
| H A D | image_ops_test.py | 1782 x_shape, argument
1801 x_shape, argument
2111 x_shape, argument
2130 x_shape, argument
2363 x_shape, argument
3299 x_shape, argument
3316 x_shape, argument
3847 def _assertResizeEqual(self, x, x_shape, y, y_shape, argument
3860 def _assertResizeCheckShape(self, x, x_shape, target_shape, argument
3935 x_shape, argument
[all …]
|
| H A D | nn_batchnorm_test.py | 298 def _testBatchNormArbitraryShapes(self, x_shape, param_shape, atol=0.0001, argument
368 def _testSuffStats(self, x_shape, axes, shift, keep_dims, has_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/framework/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_mul_op_test.cc | 38 void TestMul(const std::vector<int64_t>& x_shape, in TestMul()
87 void TestMulShape(const std::vector<int64_t>& x_shape, in TestMulShape()
137 void TimeMul(const std::vector<int64_t>& x_shape, in TimeMul()
|
| H A D | quantized_add_op_test.cc | 38 void TestAdd(const std::vector<int64_t>& x_shape, in TestAdd()
87 void TestAddShape(const std::vector<int64_t>& x_shape, in TestAddShape()
137 void TimeAdd(const std::vector<int64_t>& x_shape, in TimeAdd()
|
| H A D | betainc_op.cc | 49 const TensorShape& x_shape = x.shape(); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A 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
|
| H A D | beta_op.cc | 39 const TensorShape& x_shape = ctx->InputShape(2); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | triangular_solve_expander_test.cc | 84 auto x_shape = lx.shape(); in TEST_P() local
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | flop_counter.py | 132 def conv_flop(x_shape, w_shape, _bias, _stride, _padding, _dilation, transposed, *args, out_shape=N… argument
140 x_shape, argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_ck_backend.py | 223 def test_max_autotune_addmm(self, max_autotune_gemm_backends, x_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | atrous_convolution_test.py | 250 def _test_gradient(self, x_shape, f_shape, dilation_rate, padding): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| H A D | sparse_tensor_dense_matmul_op_test.py | 286 def _sparse_tensor_dense_vs_dense_matmul_benchmark_sparse(x_ind, x_val, x_shape, argument
|