Searched refs:array_shapes (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/pytorch/test/quantization/core/ |
H A D | test_workflow_ops.py | 257 X=hu.tensor(shapes=hu.array_shapes(1, 5,), 274 X=hu.tensor(shapes=hu.array_shapes(1, 5,), 425 @given(X=hu.tensor(shapes=hu.array_shapes(1, 5,), 438 @given(X=hu.tensor(shapes=hu.array_shapes(1, 5,), 496 @given(X=hu.tensor(shapes=hu.array_shapes(1, 5,), 507 @given(X=hu.tensor(shapes=hu.array_shapes(1, 5,), 520 X=hu.tensor(shapes=hu.array_shapes(1, 5,), 545 X=hu.tensor(shapes=hu.array_shapes(1, 5,), 681 X=hu.per_channel_tensor(shapes=hu.array_shapes(1, 5,), 758 @given(X=hu.per_channel_tensor(shapes=hu.array_shapes(1, 5,), [all …]
|
H A D | test_quantized_op.py | 302 @given(X=hu.tensor(shapes=hu.array_shapes(1, 5, 1, 5), 321 @given(X=hu.tensor(shapes=hu.array_shapes(1, 5, 1, 5), 386 @given(X=hu.tensor(shapes=hu.array_shapes(1, 5, 1, 5), 431 @given(X=hu.tensor(shapes=hu.array_shapes(1, 5, 1, 5), 457 @given(X=hu.tensor(shapes=hu.array_shapes(1, 5, 1, 5), 638 @given(X=hu.tensor(shapes=hu.array_shapes(1, 5, 1, 5), 669 @given(X=hu.tensor(shapes=hu.array_shapes(1, 5, 1, 5), 698 @given(X=hu.tensor(shapes=hu.array_shapes(1, 8, 1, 8, max_numel=10**5), 741 @given(X=hu.tensor(shapes=hu.array_shapes(1, 8, 1, 8, max_numel=10**5), 847 @given(A=hu.tensor(shapes=hu.array_shapes(1, 4, 1, 5), [all …]
|
H A D | test_workflow_module.py | 835 X=hu.per_channel_tensor(shapes=hu.array_shapes(2, 5,),
|
H A D | test_quantized_tensor.py | 1356 @given(X=hu.tensor(shapes=hu.array_shapes(min_dims=2, max_dims=4,
|
/aosp_15_r20/external/ComputeLibrary/examples/gemm_tuner/ |
H A D | cl_gemm_benchmark.sh | 385 local array_shapes 397 array_shapes=($( $shapes_list_cmd )) 443 gemm_shape=${array_shapes[$array_shapes_idx]}
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | hypothesis_utils.py | 162 def array_shapes(draw, min_dims=1, max_dims=None, min_side=1, max_side=None, max_numel=None): function
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_xnnpack_integration.py | 38 data_shape=hu.array_shapes(1, 3, 2, 64), 246 data_shape=hu.array_shapes(1, 3, 2, 64),
|
/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/ |
H A D | test_numeric.py | 1871 shape=hynp.array_shapes(),
|