| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | debug_stripper_test.cc | 126 Tensor x_t(DT_FLOAT, TensorShape({})); in TEST_F() local
185 Tensor x_t(DT_FLOAT, TensorShape({})); in TEST_F() local
223 Tensor x_t(DT_FLOAT, TensorShape({})); in TEST_F() local
|
| H A D | arithmetic_optimizer_test.cc | 1284 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 3, 28, 28})); in TEST_F() local
1323 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 3, 28, 28})); in TEST_F() local
1352 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({4, 3, 28, 28})); in TEST_F() local
1384 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({4, 3, 28, 28})); in TEST_F() local
1416 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({8, 3, 28, 28})); in TEST_F() local
1477 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({8, 3, 28, 28, 4})); in TEST_F() local
1882 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({8, 12, 28, 28})); in TEST_F() local
1920 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 3})); in TEST_F() local
2258 auto x_t = GenerateRandomTensor<DT_UINT8>(TensorShape({2, 3})); in TEST_F() local
2292 auto x_t = GenerateRandomTensor<DT_INT8>(TensorShape({2, 3})); in TEST_F() local
[all …]
|
| H A D | constant_folding_test.cc | 49 Tensor x_t(DTYPE, TensorShape({2, 2})); in SimpleNeutralElementTest() local
388 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F() local
445 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F() local
516 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F() local
972 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F() local
3396 Tensor x_t(DT_BOOL, TensorShape({})); in TEST_F() local
3652 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({})); in TEST_F() local
4003 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({1, 2, 3, 4})); in TEST_F() local
4051 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({1, 2, 3, 4})); in TEST_F() local
4129 Tensor x_t(DT_INT64, TensorShape({2, 2})); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fused_batch_norm_op_test.cc | 230 Tensor x_t(dtype, data_format == FORMAT_NHWC ? TensorShape({n, h, w, c}) in FusedBatchNormInference() local
272 Tensor x_t(dtype, shape); in FusedBatchNormGrad() local
|
| H A D | fused_batch_norm_ex_op_test.cc | 577 Tensor x_t(dtype, data_format == FORMAT_NHWC ? TensorShape({n, h, w, c}) in FusedBatchNormEx() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | SelectFixture.h | 92 TensorType x_t = create_tensor<TensorType>(shape, data_type); in compute_target() local
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_autograd_function.py | 795 def jvp(ctx, x_t): argument
811 def jvp(ctx, x_t, y_t): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | tensor_util_test.cc | 569 Tensor x_t; in CompareTensorValues() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_autograd.py | 6311 def jvp(ctx, x_t, y_t): argument
6370 def jvp(ctx, x_t, y_t): argument
8390 def jvp(ctx, x_t): argument
8693 def jvp(ctx, nt, x_t, nt2, y_t): argument
8742 def jvp(ctx, x_t, y_t): argument
8787 def jvp(ctx, x_t, y_t, _): argument
8818 def jvp(ctx, x_t): argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_eager_transforms.py | 1192 def jvp(ctx, x_t): argument
2895 def jvp_fn(x_t): argument
2914 def jvp_fn(x_t): argument
|