Home
last modified time | relevance | path

Searched defs:sizes (Results 1 – 25 of 1038) sorted by relevance

12345678910>>...42

/external/pytorch/c10/core/
DMemoryFormat.h66 inline std::vector<T> get_channels_last_strides_2d(ArrayRef<T> sizes) { in get_channels_last_strides_2d()
86 inline std::vector<int64_t> get_channels_last_strides_2d(IntArrayRef sizes) { in get_channels_last_strides_2d()
91 std::vector<T> get_channels_last_strides_3d(ArrayRef<T> sizes) { in get_channels_last_strides_3d()
113 inline std::vector<int64_t> get_channels_last_strides_3d(IntArrayRef sizes) { in get_channels_last_strides_3d()
130 const ArrayRef<T> sizes, in is_channels_last_strides_2d_s4()
172 const ArrayRef<T> sizes, in is_channels_last_strides_3d_s5()
248 const ArrayRef<T> sizes, in is_channels_last_strides_2d()
264 const ArrayRef<T> sizes, in is_channels_last_strides_3d()
279 const IntArrayRef sizes, in is_channels_last_strides_2d()
285 const IntArrayRef sizes, in is_channels_last_strides_3d()
DContiguity.h14 bool _compute_contiguous(ArrayRef<T> sizes, ArrayRef<T> strides, T numel) { in _compute_contiguous()
37 ArrayRef<T> sizes, in _compute_channels_last_contiguous_2d()
66 ArrayRef<T> sizes, in _compute_channels_last_contiguous_3d()
95 ArrayRef<T> sizes, in _compute_non_overlapping_and_dense()
/external/pytorch/aten/src/ATen/test/
Dcuda_apply_test.cpp22 int sizes[] = {4, 4}; in TEST() local
34 int sizes[] = {6, 3, 7}; in TEST() local
45 int sizes[] = {4, 3, 2}; in TEST() local
57 int sizes[] = {3, 2}; in TEST() local
69 int sizes[] = {3, 6, 5, 2}; in TEST() local
83 int sizes[] = {1, 10, 1, 5, 4}; in TEST() local
97 int sizes[] = {1, 1, 1}; in TEST() local
109 int sizes[] = {3, 6, 5, 2}; in TEST() local
125 int sizes[] = {3, 6, 5, 2}; in TEST() local
141 int sizes[] = {1, 1, 1}; in TEST() local
/external/executorch/runtime/core/portable_type/test/
Dtensor_impl_test.cpp40 SizesType sizes[2] = {3, 2}; in TEST_F() local
64 SizesType sizes[RANK] = {100, 100, 100, 100, 100}; in TEST_F() local
94 SizesType sizes[3] = {2, 0, 3}; in TEST_F() local
123 SizesType sizes[2] = {3, 2}; in TEST_F() local
151 SizesType sizes[2] = {3, 2}; in TEST_F() local
198 SizesType sizes[0] = {}; in TEST_F() local
224 SizesType sizes[2] = {3, 2}; in TEST_F() local
274 SizesType sizes[3] = {2, 3, 4}; in TEST_F() local
303 SizesType sizes[3] = {4, 4, 4}; in TEST_F() local
332 SizesType sizes[3] = {4, 4, 4}; in TEST_F() local
[all …]
Dtensor_test.cpp30 TensorImpl::SizesType sizes[1] = {1}; in TEST_F() local
45 TensorImpl::SizesType sizes[1] = {5}; in TEST_F() local
56 TensorImpl::SizesType sizes[2] = {2, 2}; in TEST_F() local
70 TensorImpl::SizesType sizes[1] = {1}; in TEST_F() local
/external/leakcanary2/shark/src/test/java/shark/internal/
DDominatorTreeTest.kt41 val sizes = tree.computeRetainedSizes(setOf(child), `10 bytes per object`) constant
50 val sizes = tree.computeRetainedSizes(setOf(root), `10 bytes per object`) constant
60 val sizes = tree.computeRetainedSizes(setOf(root), `10 bytes per object`) constant
71 val sizes = tree.computeRetainedSizes(setOf(root, child), `10 bytes per object`) constant
86 val sizes = tree.computeRetainedSizes(setOf(root, child1, child2), `10 bytes per object`) constant
102 val sizes = tree.computeRetainedSizes(setOf(root, child1, child2), `10 bytes per object`) constant
119 val sizes = constant
139 val sizes = constant
156 val sizes = constant
/external/OpenCL-CTS/test_conformance/gl/
Dtest_images_2D.cpp32 void calc_2D_test_size_descriptors(sizevec_t* sizes, size_t nsizes) in calc_2D_test_size_descriptors()
53 void calc_cube_test_size_descriptors(sizevec_t* sizes, size_t nsizes) in calc_cube_test_size_descriptors()
80 sizevec_t sizes[nsizes]; in test_images_read_2D() local
100 sizevec_t sizes[nsizes]; in test_images_read_cube() local
120 sizevec_t sizes[nsizes]; in test_images_write() local
140 sizevec_t sizes[nsizes]; in test_images_write_cube() local
159 sizevec_t sizes[nsizes]; in test_images_2D_getinfo() local
179 sizevec_t sizes[nsizes]; in test_images_cube_getinfo() local
Dtest_images_1D.cpp29 void calc_test_size_descriptors(sizevec_t* sizes, size_t nsizes) in calc_test_size_descriptors()
58 sizevec_t sizes[nsizes]; in test_images_read_1D() local
73 sizevec_t sizes[nsizes]; in test_images_write_1D() local
89 sizevec_t sizes[nsizes]; in test_images_1D_getinfo() local
106 sizevec_t sizes[nsizes]; in test_images_read_texturebuffer() local
121 sizevec_t sizes[nsizes]; in test_images_write_texturebuffer() local
137 sizevec_t sizes[nsizes]; in test_images_texturebuffer_getinfo() local
Dtest_images_depth.cpp33 void calc_depth_size_descriptors(sizevec_t* sizes, size_t nsizes) in calc_depth_size_descriptors()
53 void calc_depth_array_size_descriptors(sizevec_t* sizes, size_t nsizes) in calc_depth_array_size_descriptors()
94 sizevec_t sizes[nsizes]; in test_images_read_2D_depth() local
120 sizevec_t sizes[nsizes]; in test_images_write_2D_depth() local
142 sizevec_t sizes[nsizes]; in test_images_read_2Darray_depth() local
166 sizevec_t sizes[nsizes]; in test_images_write_2Darray_depth() local
Dtest_images_3D.cpp32 void calc_3D_size_descriptors(sizevec_t* sizes, size_t nsizes) in calc_3D_size_descriptors()
61 sizevec_t sizes[nsizes]; in test_images_read_3D() local
90 sizevec_t sizes[nsizes]; in test_images_write_3D() local
109 sizevec_t sizes[nsizes]; in test_images_3D_getinfo() local
Dtest_images_multisample.cpp30 void calc_2D_multisample_size_descriptors(sizevec_t* sizes, size_t nsizes) in calc_2D_multisample_size_descriptors()
49 void calc_2D_array_multisample_size_descriptors(sizevec_t* sizes, size_t nsizes) in calc_2D_array_multisample_size_descriptors()
83 sizevec_t sizes[nsizes]; in test_images_read_2D_multisample() local
118 sizevec_t sizes[nsizes]; in test_images_read_2Darray_multisample() local
Dtest_images_1Darray.cpp28 void calc_1D_array_size_descriptors(sizevec_t* sizes, size_t nsizes) in calc_1D_array_size_descriptors()
57 sizevec_t sizes[nsizes]; in test_images_read_1Darray() local
72 sizevec_t sizes[nsizes]; in test_images_write_1Darray() local
88 sizevec_t sizes[nsizes]; in test_images_1Darray_getinfo() local
Dtest_images_2Darray.cpp29 void calc_2D_array_size_descriptors(sizevec_t* sizes, size_t nsizes) in calc_2D_array_size_descriptors()
59 sizevec_t sizes[nsizes]; in test_images_read_2Darray() local
76 sizevec_t sizes[nsizes]; in test_images_write_2Darray() local
92 sizevec_t sizes[nsizes]; in test_images_2Darray_getinfo() local
/external/executorch/backends/vulkan/runtime/graph/ops/glsl/
Dindexing_utils.h81 ivec4 tidx_to_nchwi(const ivec4 tidx, const ivec4 sizes, const int packed_dim) { in tidx_to_nchwi()
106 ivec4 nchwi_to_tidx(const int nchwi, const ivec4 sizes) { in nchwi_to_tidx()
114 int tidx_to_nchwi(const ivec4 tidx, const ivec4 sizes) { in tidx_to_nchwi()
149 ivec4 sizes, in lpos_to_tidx()
168 ivec4 sizes, in tidx_to_lpos()
188 ivec4 sizes, in tidx_to_pos()
211 ivec4 sizes, in tidx_to_posi()
294 bool pos_out_of_bounds(ivec3 pos, ivec4 sizes, int packed_dim) { in pos_out_of_bounds()
310 ivec4 to_tensor_idx(ivec3 pos, ivec4 sizes, int packed_dim) { in to_tensor_idx()
339 ivec3 to_texture_pos(ivec4 idx, ivec4 sizes, int packed_dim) { in to_texture_pos()
[all …]
/external/executorch/extension/tensor/
Dtensor_ptr_maker.cpp80 std::vector<exec_aten::SizesType> sizes, in random_strided()
100 std::vector<exec_aten::SizesType> sizes, in empty_strided()
117 std::vector<exec_aten::SizesType> sizes, in full_strided()
136 std::vector<exec_aten::SizesType> sizes, in rand_strided()
149 std::vector<exec_aten::SizesType> sizes, in randn_strided()
164 std::vector<exec_aten::SizesType> sizes, in randint_strided()
Dtensor_ptr.cpp30 std::vector<exec_aten::SizesType> sizes; member
58 std::vector<exec_aten::SizesType> sizes, in make_tensor_ptr()
131 std::vector<exec_aten::SizesType> sizes, in make_tensor_ptr()
154 std::vector<exec_aten::SizesType> sizes( in clone_tensor_ptr() local
188 const std::vector<exec_aten::SizesType>& sizes) { in resize_tensor_ptr()
/external/executorch/kernels/test/
Dop_scalar_tensor_test.cpp36 std::vector<int32_t> sizes{}; in test_scalar_tensor_out_0d() local
49 std::vector<int32_t> sizes{1}; in test_scalar_tensor_out_1d() local
59 std::vector<int32_t> sizes{1, 1}; in test_scalar_tensor_out_2d() local
69 std::vector<int32_t> sizes{1, 1, 1}; in test_scalar_tensor_out_3d() local
109 std::vector<int32_t> sizes{1, 2, 1}; local
Dop_gather_test.cpp42 const std::vector<int32_t> sizes = {2, 3}; in test_gather_out() local
145 const std::vector<int32_t> sizes = {2, 3}; in test_gather_out_invalid_dim() local
212 const std::vector<int32_t> sizes = {2, 3}; in TEST_F() local
243 const std::vector<int32_t> sizes = {2, 3}; in TEST_F() local
283 const std::vector<int32_t> sizes = {2, 0, 3}; in TEST_F() local
308 const std::vector<int32_t> sizes = {2, 3}; in TEST_F() local
321 const std::vector<int32_t> sizes = {}; in TEST_F() local
334 const std::vector<int32_t> sizes = {3}; in TEST_F() local
347 const std::vector<int32_t> sizes = {}; in TEST_F() local
360 const std::vector<int32_t> sizes = {3}; in TEST_F() local
[all …]
Dop_constant_pad_nd_test.cpp41 const std::vector<int32_t> sizes = {2, 4, 4}; in test_constant_pad_nd_out_dim2() local
89 const std::vector<int32_t> sizes = {2, 4, 4}; in test_constant_pad_nd_out_dim1() local
141 const std::vector<int32_t> sizes = {2, 4, 4}; in test_constant_pad_nd_out_dim0() local
194 const std::vector<int32_t> sizes = {2, 4, 4}; in test_constant_pad_nd_out_dim12() local
246 const std::vector<int32_t> sizes = {2, 4, 4}; in test_constant_pad_nd_out_dim02() local
299 const std::vector<int32_t> sizes = {2, 4, 4}; in test_constant_pad_nd_out_dim012() local
404 const std::vector<int32_t> sizes = {1, 4, 4}; in TEST_F() local
420 const std::vector<int32_t> sizes = {1, 4, 4}; in TEST_F() local
436 const std::vector<int32_t> sizes = {1, 4, 4}; in TEST_F() local
456 const std::vector<int32_t> sizes = {1, 4, 4}; in TEST_F() local
Dop_zeros_test.cpp34 std::vector<int64_t> sizes(size_int32_t.begin(), size_int32_t.end()); in test_zeros_out() local
57 int64_t sizes[2] = {3, 2}; in ET_FORALL_REAL_TYPES_AND() local
69 int64_t sizes[2] = {3, 2}; in TEST_F() local
84 int64_t sizes[2] = {3, 2}; in TEST_F() local
Dop_full_test.cpp31 op_full_out(const IntArrayRef sizes, const Scalar& fill_value, Tensor& out) { in op_full_out()
83 auto sizes = IntArrayRef(sizes_int64_t_vec.data(), sizes_int64_t_vec.size()); in TEST_F() local
95 auto sizes = IntArrayRef(sizes_int64_t_vec.data(), sizes_int64_t_vec.size()); in TEST_F() local
118 auto sizes = IntArrayRef(sizes_int64_t_vec.data(), sizes_int64_t_vec.size()); in TEST_F() local
/external/executorch/runtime/core/exec_aten/testing_util/
Dtensor_factory.h32 inline size_t sizes_to_numel(const std::vector<int32_t>& sizes) { in sizes_to_numel()
56 const std::vector<int32_t> sizes, in check_strides()
128 const std::vector<int32_t>& sizes, in check_dim_order()
151 const std::vector<int32_t>& sizes, in strides_from_dim_order()
/external/executorch/runtime/core/exec_aten/testing_util/test/
Dtensor_factory_test.cpp124 std::vector<int64_t> sizes = {2, 2}; in TEST_F() local
130 int32_t sizes[dim] = {2, 2}; in TEST_F() local
153 std::vector<int64_t> sizes = {2, 2}; in TEST_F() local
159 int32_t sizes[dim] = {2, 2}; in TEST_F() local
183 std::vector<int64_t> sizes = {2, 2}; in TEST_F() local
189 int32_t sizes[dim] = {2, 2}; in TEST_F() local
310 std::vector<int64_t> sizes = {2, 2}; in TEST_F() local
322 int32_t sizes[dim] = {2, 2}; in TEST_F() local
347 std::vector<int64_t> sizes = {2, 2}; in TEST_F() local
359 int32_t sizes[dim] = {2, 2}; in TEST_F() local
[all …]
/external/executorch/kernels/portable/test/
Dop_div_test.cpp56 const std::vector<int32_t> sizes = {2, 2}; in TEST_F() local
74 const std::vector<int32_t> sizes = {2, 2}; in TEST_F() local
91 const std::vector<int32_t> sizes = {2, 2}; in TEST_F() local
108 const std::vector<int32_t> sizes = {2, 2}; in TEST_F() local
/external/pytorch/aten/src/ATen/core/
Dtensor_type.cpp14 bool possible_cross_dimension_overlap(c10::IntArrayRef sizes, c10::IntArrayRef strides) { in possible_cross_dimension_overlap()
106 auto sizes = ss.sizes().value(); in operator <<() local
151 at::IntArrayRef sizes, in computeStrideProps()
259 VaryingShape<int64_t> sizes; in create() local
279 const VaryingShape<int64_t>& sizes, in create()
309 const SymbolicShape& sizes, in create()
343 VaryingShape<int64_t> TensorType::sizes() const { in sizes() function in c10::TensorType
436 SymbolicShape sizes, in TensorType()
451 at::IntArrayRef sizes) { in createContiguous()

12345678910>>...42