Home
last modified time | relevance | path

Searched refs:array_shapes (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_workflow_ops.py257 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 Dtest_quantized_op.py302 @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 Dtest_workflow_module.py835 X=hu.per_channel_tensor(shapes=hu.array_shapes(2, 5,),
H A Dtest_quantized_tensor.py1356 @given(X=hu.tensor(shapes=hu.array_shapes(min_dims=2, max_dims=4,
/aosp_15_r20/external/ComputeLibrary/examples/gemm_tuner/
H A Dcl_gemm_benchmark.sh385 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 Dhypothesis_utils.py162 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 Dtest_xnnpack_integration.py38 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 Dtest_numeric.py1871 shape=hynp.array_shapes(),