Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/lib/
H A Dtest_shape_base_.py26 array_split,
45 array_split,
337 assert_raises(ValueError, array_split, a, 0)
341 res = array_split(a, 1)
345 res = array_split(a, 2)
349 res = array_split(a, 3)
353 res = array_split(a, 4)
357 res = array_split(a, 5)
367 res = array_split(a, 6)
378 res = array_split(a, 7)
[all …]
/aosp_15_r20/external/mesa3d/src/compiler/nir/
H A Dnir_split_vars.c400 struct array_split { struct
405 struct array_split *splits; argument
414 struct array_split root_split;
553 struct array_split *split, in create_split_array_vars()
582 split->splits = rzalloc_array(mem_ctx, struct array_split, in create_split_array_vars()
867 struct array_split *split = &info->root_split; in split_array_access_impl()
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/saving/
H A Dhdf5_format.py830 chunked_data = np.array_split(data_npy, num_chunks)
835 chunked_data = np.array_split(data_npy, num_chunks)
/aosp_15_r20/external/pytorch/test/
H A Dtest_view_ops.py1848 result_n = np.array_split(a_n, sections, dim)
1895 result_n = np.array_split(a_n, indices, dim)
1949 np.array_split(a.cpu().numpy(), sections_or_indices, dim)
H A Dtest_tensorexpr.py1067 npr_a, npr_b = np.array_split(npr2, 2)
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dparameter_server_strategy_v2_test.py410 expected_result = np.array_split(expected_result, 2)
/aosp_15_r20/external/pytorch/torch/_numpy/
H A D_funcs_impl.py257 def array_split(ary: ArrayLike, indices_or_sections, axis=0): function
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py17536 ref=np.array_split,