Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
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 …]
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
Dsplit_v_op_test.cc110 #define BM_SPLITV_3D(split_dim, num_split, dim0, dim1, dim2) \ argument
/external/aac/libFDK/src/
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()
Dfft.cpp1578 const int dim1, const int dim2, in fftN2_func()
1693 #define fftN2(DATA_TYPE, pInput, length, dim1, dim2, fft_func1, fft_func2, \ argument
Dnlc_dec.cpp389 const INT dim1, const INT dim2, SCHAR out_data[][2], in huff_dec_2D()
/external/aac/libFDK/include/
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
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
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()
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()
/external/eigen/unsupported/test/
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()
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
Dcxx11_tensor_reverse_sycl.cpp26 IndexType dim2 = 3; in test_simple_reverse() local
117 IndexType dim2 = 3; in test_expr_reverse() local
Dcxx11_tensor_morphing_sycl.cpp35 typename Tensor<DataType, 3 ,DataLayout, IndexType>::Dimensions dim2(2,3,7); in test_simple_reshape() local
93 typename Tensor<DataType, 2, DataLayout, IndexType>::Dimensions dim2(6,7); in test_reshape_as_lvalue() local
Dcxx11_tensor_block_access.cpp531 for (int dim2 = 0; dim2 < 3; ++dim2) { in test_empty_dims() local
Dcxx11_tensor_morphing.cpp28 Tensor<float, 2>::Dimensions dim2(6,7); in test_simple_reshape() local
/external/ComputeLibrary/tests/validation/reference/
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
/external/webrtc/modules/audio_coding/codecs/ilbc/
Dmy_corr.c32 size_t dim2 /* (i) dimension seq2 */ in WebRtcIlbcfix_MyCorr()
/external/ComputeLibrary/arm_compute/core/
DValidate.h47 inline bool have_different_dimensions(const Dimensions<T> &dim1, const Dimensions<T> &dim2, unsigne… in have_different_dimensions()
/external/tensorflow/tensorflow/lite/kernels/shim/
Dshape.cc73 int Shape::AddDims(const int dim1, const int dim2) { in AddDims()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h447 int64_t dim2 = 0; in CreateR3Projected() local
472 int64_t dim2 = 0; in CreateR4Projected() local
/external/armnn/src/backends/reference/workloads/
DSlice.cpp53 unsigned int dim2 = paddedInput[2]; in Slice() local
DStridedSlice.cpp137 int dim2 = armnn::numeric_cast<int>(inputShape[2]); in StridedSlice() local
/external/tensorflow/tensorflow/core/util/
Dmkl_util_test.cc74 memory::dims dim2 = {3, 4}; in TEST() local
/external/libxaac/decoder/
Dixheaacd_fft_ifft_32x32.c196 VOID ixheaacd_hbe_apply_tw_mult_ifft(FLOAT32 *inp, FLOAT32 *op, WORD32 dim1, WORD32 dim2, in ixheaacd_hbe_apply_tw_mult_ifft()
224 VOID ixheaacd_hbe_apply_tw_mult_fft(FLOAT32 *inp, FLOAT32 *op, WORD32 dim1, WORD32 dim2, in ixheaacd_hbe_apply_tw_mult_fft()
/external/tensorflow/tensorflow/core/ir/types/
Ddialect.cc492 int64_t dim2 = std::get<1>(dims); in GetCastCompatibleShape() local

12