Home
last modified time | relevance | path

Searched defs:dim2 (Results 1 – 25 of 100) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_op_test.cc61 int concat_dimension, int dim2, in ConcatHelper()
90 const int dim2 = state.range(0); in BM_ConcatDim0Float() local
96 const int dim2 = state.range(0); in BM_ConcatDim1Float() local
113 const int dim2 = state.range(0); in BM_ConcatDim0String() local
126 const int dim2 = state.range(0); in BM_ConcatDim1uint8() local
131 const int dim2 = state.range(0); in BM_ConcatDim1int16() local
136 const int dim2 = state.range(0); in BM_ConcatDim1bfloat16() local
159 int concat_dimension, int dim2) { in ConcatManyHelper()
188 const int dim2 = state.range(0); in BM_ConcatManyDim1bfloat16() local
195 void MemcpyAlternativeHelper(::testing::benchmark::State& state, int dim2) { in MemcpyAlternativeHelper()
[all …]
H A Dquantized_concat_op_test.cc252 int concat_dimension, bool same_limits, int dim2) { in ConcatHelper()
286 const int dim2 = state.range(0); in BM_QConcatDim0SameLimitQInt32() local
293 const int dim2 = state.range(0); in BM_QConcatDim1SameLimitQInt32() local
301 const int dim2 = state.range(0); in BM_QConcatDim0DifferLimitQInt32() local
309 const int dim2 = state.range(0); in BM_QConcatDim1DifferLimitQInt32() local
337 const int dim2 = state.range(0); in BM_QConcatDim0SameLimitQUint8() local
344 const int dim2 = state.range(0); in BM_QConcatDim1SameLimitQUint8() local
352 const int dim2 = state.range(0); in BM_QConcatDim0DifferLimitQUint8() local
360 const int dim2 = state.range(0); in BM_QConcatDim1DifferLimitQUint8() local
/aosp_15_r20/external/aac/libFDK/src/
H A DFDK_matrixCalloc.cpp134 void **fdkCallocMatrix2D(UINT dim1, UINT dim2, UINT size) { in fdkCallocMatrix2D()
155 void **fdkCallocMatrix2D_aligned(UINT dim1, UINT dim2, UINT size) { in fdkCallocMatrix2D_aligned()
188 void **fdkCallocMatrix2D_int(UINT dim1, UINT dim2, UINT size, in fdkCallocMatrix2D_int()
211 void **fdkCallocMatrix2D_int_aligned(UINT dim1, UINT dim2, UINT size, in fdkCallocMatrix2D_int_aligned()
236 void ***fdkCallocMatrix3D(UINT dim1, UINT dim2, UINT dim3, UINT size) { in fdkCallocMatrix3D()
278 void ***fdkCallocMatrix3D_int(UINT dim1, UINT dim2, UINT dim3, UINT size, in fdkCallocMatrix3D_int()
/aosp_15_r20/external/aac/libFDK/include/
H A DFDK_matrixCalloc.h182 #define FDK_ALLOCATE_MEMORY_2D(a, dim1, dim2, type) \ argument
188 #define FDK_ALLOCATE_MEMORY_2D_INT(a, dim1, dim2, type, s) \ argument
194 #define FDK_ALLOCATE_MEMORY_2D_INT_ALIGNED(a, dim1, dim2, type, s) \ argument
212 #define FDK_ALLOCATE_MEMORY_3D(a, dim1, dim2, dim3, type) \ argument
218 #define FDK_ALLOCATE_MEMORY_3D_INT(a, dim1, dim2, dim3, type, s) \ argument
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorMap.h98 …EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, Index dim1, Index dim2) : m_data(dataPtr… in TensorMap()
102 …EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, Index dim1, Index dim2, Index dim3) : m_… in TensorMap()
106 …EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, Index dim1, Index dim2, Index dim3, Inde… in TensorMap()
110 …EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, Index dim1, Index dim2, Index dim3, Inde… in TensorMap()
H A DTensor.h353 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2) in Tensor()
358 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2, Index dim3) in Tensor()
363 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2, Index dim3, Index dim4) in Tensor()
368 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2, Index dim3, Index dim4, Index… in Tensor()
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_index_list.cpp272 int dim2 = 1; in test_dynamic_index_list() local
303 int dim2 = 1; in test_mixed_index_list() local
367 Eigen::IndexList<Eigen::type2index<1>, int> dim2; in test_dim_check() local
H A Dcxx11_tensor_ifft.cpp68 static void test_3D_fft_ifft_invariant(int dim0, int dim1, int dim2) { in test_3D_fft_ifft_invariant()
100 static void test_sub_fft_ifft_invariant(int dim0, int dim1, int dim2, int dim3) { in test_sub_fft_ifft_invariant()
H A Dcxx11_tensor_reverse_sycl.cpp26 IndexType dim2 = 3; in test_simple_reverse() local
117 IndexType dim2 = 3; in test_expr_reverse() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_diagonal_copy.cpp24 int64_t dim2, in diagonal_copy_impl()
69 int64_t dim2, in diagonal_copy_out()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DFuseBatchNormalization.cpp47 const unsigned int dim2 = w.shape()[2]; in fuse_batch_normalization_dwc_layer() local
83 const unsigned int dim2 = w.shape()[2]; in fuse_batch_normalization_conv_layer() local
/aosp_15_r20/external/guava/android/guava-tests/test/com/google/common/primitives/
H A DBytesTest.java137 int dim2 = 1 << 15; in testConcat_overflow_negative() local
145 int dim2 = 1 << 16; in testConcat_overflow_nonNegative() local
H A DBooleansTest.java136 int dim2 = 1 << 15; in testConcat_overflow_negative() local
144 int dim2 = 1 << 16; in testConcat_overflow_nonNegative() local
H A DCharsTest.java228 int dim2 = 1 << 15; in testConcat_overflow_negative() local
236 int dim2 = 1 << 16; in testConcat_overflow_nonNegative() local
H A DShortsTest.java249 int dim2 = 1 << 15; in testConcat_overflow_negative() local
257 int dim2 = 1 << 16; in testConcat_overflow_nonNegative() local
H A DDoublesTest.java286 int dim2 = 1 << 15; in testConcat_overflow_negative() local
294 int dim2 = 1 << 16; in testConcat_overflow_nonNegative() local
/aosp_15_r20/external/guava/guava-tests/test/com/google/common/primitives/
H A DBytesTest.java137 int dim2 = 1 << 15; in testConcat_overflow_negative() local
145 int dim2 = 1 << 16; in testConcat_overflow_nonNegative() local
H A DBooleansTest.java136 int dim2 = 1 << 15; in testConcat_overflow_negative() local
144 int dim2 = 1 << 16; in testConcat_overflow_nonNegative() local
H A DCharsTest.java228 int dim2 = 1 << 15; in testConcat_overflow_negative() local
236 int dim2 = 1 << 16; in testConcat_overflow_nonNegative() local
H A DShortsTest.java249 int dim2 = 1 << 15; in testConcat_overflow_negative() local
257 int dim2 = 1 << 16; in testConcat_overflow_nonNegative() local
H A DIntsTest.java230 int dim2 = 1 << 15; in testConcat_overflow_negative() local
238 int dim2 = 1 << 16; in testConcat_overflow_nonNegative() local
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DTensorAccessor.h213 index_t dim2) const { in transpose()
254 index_t dim2) const { in transpose()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesViews.cpp310 int64_t offset, int64_t dim1, int64_t dim2) in diagonal_batching_rule()
322 c10::SymIntArrayRef input_sizes, int64_t offset, int64_t dim1, int64_t dim2) { in diagonal_backward_batch_rule()
520 …const Tensor& self, std::optional<int64_t> self_bdim, int64_t offset, int64_t dim1, int64_t dim2) { in diag_embed_batch_rule()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral_util.h447 int64_t dim2 = 0; in CreateR3Projected() local
472 int64_t dim2 = 0; in CreateR4Projected() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DDistance.cpp110 auto dim2 = x2.dim(); in cdist_impl() local
198 auto dim2 = _x2.dim(); in _cdist_backward() local

1234