Home
last modified time | relevance | path

Searched refs:test_shape (Results 1 – 17 of 17) sorted by relevance

/external/pytorch/test/
Dtest_sparse.py289 def test_shape(sparse_dims, nnz, with_size): function
309 test_shape(3, 10, 100)
310 test_shape(3, 10, [100, 100, 100])
311 test_shape(3, 10, [100, 100, 100, 5, 5, 5, 0])
312 test_shape(3, 0, [0, 0, 100, 5, 5, 5, 0])
811 def test_shape(sparse_dims, nnz, with_size): function
822 test_shape(4, 20, 5)
823 test_shape(3, 10, [100, 100, 100, 5, 5, 5, 0])
824 test_shape(3, 0, [0, 0, 100, 5, 5, 5, 0])
930 def test_shape(sparse_dims, nnz, with_size): function
[all …]
Dtest_sparse_csr.py1839 def test_shape(di, dj, dk, nnz0=None, nnz1=None): function
1931 test_shape(i, j, k)
1932 test_shape(4, 4, 4, 0, 0)
1941 def test_shape(d1, d2, d3, nnz, transposed, index_dtype): function
1951 test_shape(7, 8, 9, 20, False, index_dtype)
1952 test_shape(7, 8, 9, 20, True, index_dtype)
1960 def test_shape(m, n, p, nnz, broadcast, index_dtype, alpha_beta=None): function
1978 test_shape(7, 8, 9, 20, False, index_dtype, None)
1979 test_shape(7, 8, 9, 20, True, index_dtype, None)
1980 test_shape(7, 8, 9, 20, False, index_dtype, (1, 0))
[all …]
Dtest_nn.py5534 def test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners): function
5614 test_shape(N, C, H, W, H, W, mode, padding_mode, align_corners)
5623 test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners)
5632 test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners)
5641 test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners)
5649 test_shape(N, C, IH, IW, 0, W, mode, padding_mode, align_corners)
5657 test_shape(N, 0, IH, IW, H, W, mode, padding_mode, align_corners)
5665 test_shape(0, C, IH, IW, H, W, mode, padding_mode, align_corners)
5895 def test_shape(N, C, ID, IH, IW, D, H, W, mode, padding_mode, align_corners): function
5931 test_shape(N, C, D, H, W, D, H, W, mode, padding_mode, align_corners)
[all …]
Dtest_mps.py10607 def test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners): function
10683 test_shape(N, C, H, W, H, W, mode, padding_mode, align_corners)
10692 test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners)
10701 test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners)
10710 test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners)
10718 test_shape(N, C, IH, IW, 0, W, mode, padding_mode, align_corners)
10726 test_shape(N, 0, IH, IW, H, W, mode, padding_mode, align_corners)
10734 test_shape(0, C, IH, IW, H, W, mode, padding_mode, align_corners)
Dtest_autograd.py3917 def test_shape(self): member in TestAutograd
/external/tensorflow/tensorflow/core/kernels/
Dsparse_utils_test.cc352 for (const TensorShape& test_shape : kTensorShapes) { in TEST_P() local
354 GenerateRandomSparseTensor(kNumNonZeros, test_shape, ordered, indices, in TEST_P()
455 for (const TensorShape& test_shape : kTensorShapes) { in TEST_P() local
457 GenerateRandomSparseTensor(kNumNonZeros, test_shape, ordered, indices, in TEST_P()
467 for (int64_t val : {static_cast<int64_t>(-1), test_shape.dim_size(dim)}) { in TEST_P()
495 for (const TensorShape& test_shape : kTensorShapes) { in TEST_P() local
497 GenerateRandomSparseTensor(kNumNonZeros, test_shape, ordered, indices, in TEST_P()
/external/tensorflow/tensorflow/python/kernel_tests/image_ops/
Dextract_image_patches_grad_test.py151 test_shape = [
155 in_val = array_ops.placeholder(shape=test_shape, dtype=dtypes.float32)
/external/harfbuzz_ng/test/api/
Dtest-shape.c112 test_shape (void) in test_shape() function
201 hb_test_add (test_shape); in main()
/external/pytorch/test/cpp/lazy/
DCMakeLists.txt11 ${LAZY_TEST_ROOT}/test_shape.cpp
/external/tensorflow/tensorflow/lite/kernels/
Dsub_test.cc424 const std::vector<int> test_shape = {18}; in TEST() local
425 QuantizedSubOpModel m({TensorType_INT16, test_shape, -2.0, 2.0}, in TEST()
426 {TensorType_INT16, test_shape, -1.1, 1.1}, in TEST()
/external/pytorch/test/torch_np/
Dtest_basic.py263 def test_shape(self, func): member in TestShapeLikeToArray
/external/tensorflow/tensorflow/python/ops/parallel_for/
Darray_test.py91 def test_shape(self): member in ArrayTest
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_array_ops_test.py274 def test_shape(self): member in StructuredArrayOpsTest
/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
Dtf-to-tosa-pipeline.mlir597 // CHECK-LABEL: test_shape
599 func.func @test_shape() -> tensor<3xi32> {
Dtfl-to-tosa-pipeline.mlir932 // CHECK-LABEL: test_shape
934 func.func @test_shape() -> tensor<3xi32> {
/external/pytorch/test/export/
Dtest_serialize.py750 def test_shape(self): member in TestDeserialize
/external/pytorch/test/functorch/
Dtest_eager_transforms.py305 def test_shape(self, device): member in TestGradTransform