Home
last modified time | relevance | path

Searched refs:dim2 (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dconcat_op_test.cc41 static void ConcatHelper(int iters, int concat_dimension, int dim2) { in ConcatHelper() argument
49 Tensor in0(dt, TensorShape({kDim1, dim2})); in ConcatHelper()
51 Tensor in1(dt, TensorShape({kDim1, dim2})); in ConcatHelper()
64 ((kDim1 * dim2) + (kDim1 * dim2)) * sizeof(T)); in ConcatHelper()
70 static void BM_ConcatDim0Float(int iters, int dim2) { in BM_ConcatDim0Float() argument
71 ConcatHelper<float>(iters, 0, dim2); in BM_ConcatDim0Float()
74 static void BM_ConcatDim1Float(int iters, int dim2) { in BM_ConcatDim1Float() argument
75 ConcatHelper<float>(iters, 1, dim2); in BM_ConcatDim1Float()
81 static void BM_ConcatDim1uint8(int iters, int dim2) { in BM_ConcatDim1uint8() argument
82 ConcatHelper<uint8>(iters, 1, dim2); in BM_ConcatDim1uint8()
[all …]
Dquantized_concat_op_test.cc252 int dim2) { in ConcatHelper() argument
258 TensorShape shape({kDim1, dim2}); in ConcatHelper()
282 ((kDim1 * dim2) + (kDim1 * dim2)) * sizeof(T)); in ConcatHelper()
288 static void BM_QConcatDim0SameLimitQInt32(int iters, int dim2) { in BM_QConcatDim0SameLimitQInt32() argument
290 dim2); in BM_QConcatDim0SameLimitQInt32()
293 static void BM_QConcatDim1SameLimitQInt32(int iters, int dim2) { in BM_QConcatDim1SameLimitQInt32() argument
295 dim2); in BM_QConcatDim1SameLimitQInt32()
298 static void BM_QConcatDim0DifferLimitQInt32(int iters, int dim2) { in BM_QConcatDim0DifferLimitQInt32() argument
300 dim2); in BM_QConcatDim0DifferLimitQInt32()
303 static void BM_QConcatDim1DifferLimitQInt32(int iters, int dim2) { in BM_QConcatDim1DifferLimitQInt32() argument
[all …]
Dstrided_slice_op_test.cc79 static void BM_SliceFloat(int iters, int dim2) { in BM_SliceFloat() argument
80 SliceHelper<float>(iters, dim2); in BM_SliceFloat()
85 static void BM_SliceComplex64(int iters, int dim2) { in BM_SliceComplex64() argument
86 SliceHelper<std::complex<float>>(iters, dim2); in BM_SliceComplex64()
91 static void BM_SliceBFloat16(int iters, int dim2) { in BM_SliceBFloat16() argument
92 SliceHelper<bfloat16>(iters, dim2); in BM_SliceBFloat16()
Dslice_op_test.cc72 static void BM_SliceFloat(int iters, int dim2) { in BM_SliceFloat() argument
73 SliceHelper<float>(iters, dim2); in BM_SliceFloat()
78 static void BM_SliceBFloat16(int iters, int dim2) { in BM_SliceBFloat16() argument
79 SliceHelper<bfloat16>(iters, dim2); in BM_SliceBFloat16()
/external/aac/libFDK/src/
DFDK_matrixCalloc.cpp134 void **fdkCallocMatrix2D(UINT dim1, UINT dim2, UINT size) { in fdkCallocMatrix2D() argument
138 if (!dim1 || !dim2) return NULL; in fdkCallocMatrix2D()
142 if ((p2 = (char *)fdkCallocMatrix1D(dim1 * dim2, size)) == NULL) { in fdkCallocMatrix2D()
149 p2 += dim2 * size; in fdkCallocMatrix2D()
155 void **fdkCallocMatrix2D_aligned(UINT dim1, UINT dim2, UINT size) { in fdkCallocMatrix2D_aligned() argument
159 if (!dim1 || !dim2) return NULL; in fdkCallocMatrix2D_aligned()
163 if ((p2 = (char *)fdkCallocMatrix1D_aligned(dim1 * dim2, size)) == NULL) { in fdkCallocMatrix2D_aligned()
170 p2 += dim2 * size; in fdkCallocMatrix2D_aligned()
188 void **fdkCallocMatrix2D_int(UINT dim1, UINT dim2, UINT size, in fdkCallocMatrix2D_int() argument
194 if (!dim1 || !dim2) return NULL; in fdkCallocMatrix2D_int()
[all …]
Dnlc_dec.cpp389 const INT dim1, const INT dim2, SCHAR out_data[][2], in huff_dec_2D() argument
455 (HANDLE_HUFF_NODE)&FDK_huffCLDNodes.h2D[dim1][dim2]->lav3[0][0]; in huff_dec_2D()
459 (HANDLE_HUFF_NODE)&FDK_huffCLDNodes.h2D[dim1][dim2]->lav5[0][0]; in huff_dec_2D()
463 (HANDLE_HUFF_NODE)&FDK_huffCLDNodes.h2D[dim1][dim2]->lav7[0][0]; in huff_dec_2D()
467 (HANDLE_HUFF_NODE)&FDK_huffCLDNodes.h2D[dim1][dim2]->lav9[0][0]; in huff_dec_2D()
475 (HANDLE_HUFF_NODE)&FDK_huffICCNodes.h2D[dim1][dim2]->lav1[0][0]; in huff_dec_2D()
479 (HANDLE_HUFF_NODE)&FDK_huffICCNodes.h2D[dim1][dim2]->lav3[0][0]; in huff_dec_2D()
483 (HANDLE_HUFF_NODE)&FDK_huffICCNodes.h2D[dim1][dim2]->lav5[0][0]; in huff_dec_2D()
487 (HANDLE_HUFF_NODE)&FDK_huffICCNodes.h2D[dim1][dim2]->lav7[0][0]; in huff_dec_2D()
494 nodeTab = (HANDLE_HUFF_NODE)&huffOLDNodes.h2D[dim1][dim2]->lav3[0][0]; in huff_dec_2D()
[all …]
Dfft.cpp1578 const int dim1, const int dim2, in fftN2_func() argument
1591 FDK_ASSERT(length == dim1 * dim2); in fftN2_func()
1600 for (i = 0; i < dim2; i++) { in fftN2_func()
1602 pDst[2 * j] = pSrc[2 * j * dim2]; in fftN2_func()
1603 pDst[2 * j + 1] = pSrc[2 * j * dim2 + 1]; in fftN2_func()
1653 for (int j = 0; j < dim2; j++) { in fftN2_func()
1661 switch (dim2) { in fftN2_func()
1683 for (int j = 0; j < dim2; j++) { in fftN2_func()
1693 #define fftN2(DATA_TYPE, pInput, length, dim1, dim2, fft_func1, fft_func2, \ argument
1697 C_AALLOC_SCRATCH_START(aDst2, DATA_TYPE, 2 * dim2) \
[all …]
/external/aac/libFDK/include/
DFDK_matrixCalloc.h123 void** fdkCallocMatrix2D(UINT dim1, UINT dim2, UINT size);
124 void** fdkCallocMatrix2D_aligned(UINT dim1, UINT dim2, UINT size);
127 void** fdkCallocMatrix2D_int(UINT dim1, UINT dim2, UINT size, MEMORY_SECTION s);
130 void** fdkCallocMatrix2D_int_aligned(UINT dim1, UINT dim2, UINT size,
140 void*** fdkCallocMatrix3D(UINT dim1, UINT dim2, UINT dim3, UINT size);
143 void*** fdkCallocMatrix3D_int(UINT dim1, UINT dim2, UINT dim3, UINT size,
182 #define FDK_ALLOCATE_MEMORY_2D(a, dim1, dim2, type) \ argument
183 if (((a) = (type**)fdkCallocMatrix2D((dim1), (dim2), sizeof(type))) == \
188 #define FDK_ALLOCATE_MEMORY_2D_INT(a, dim1, dim2, type, s) \ argument
189 if (((a) = (type**)fdkCallocMatrix2D_int((dim1), (dim2), sizeof(type), \
[all …]
/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() argument
69 Tensor<double, 3, DataLayout> tensor(dim0, dim1, dim2); in test_3D_fft_ifft_invariant()
85 VERIFY_IS_EQUAL(tensor_after_fft.dimension(2), dim2); in test_3D_fft_ifft_invariant()
88 VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(2), dim2); in test_3D_fft_ifft_invariant()
92 for (int k = 0; k < dim2; ++k) { 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() argument
101 Tensor<double, 4, DataLayout> tensor(dim0, dim1, dim2, dim3); in test_sub_fft_ifft_invariant()
116 VERIFY_IS_EQUAL(tensor_after_fft.dimension(2), dim2); in test_sub_fft_ifft_invariant()
120 VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(2), dim2); in test_sub_fft_ifft_invariant()
125 for (int k = 0; k < dim2; ++k) { in test_sub_fft_ifft_invariant()
Dcxx11_tensor_index_list.cpp273 int dim2 = 1; in test_dynamic_index_list() local
276 auto reduction_axis = make_index_list(dim1, dim2, dim3); in test_dynamic_index_list()
304 int dim2 = 1; in test_mixed_index_list() local
307 auto reduction_axis = make_index_list(0, dim2, 2, dim4); in test_mixed_index_list()
368 Eigen::IndexList<Eigen::type2index<1>, int> dim2; in test_dim_check() local
369 dim2.set(1, 2); in test_dim_check()
370 VERIFY(dimensions_match(dim1, dim2)); in test_dim_check()
Dcxx11_tensor_morphing.cpp28 Tensor<float, 2>::Dimensions dim2(6,7); in test_simple_reshape() local
29 tensor3 = tensor1.reshape(dim2); in test_simple_reshape()
/external/ply/ply/example/BASIC/
Dbasinterp.py91 var, dim1, dim2 = expr[1]
92 if not dim1 and not dim2:
100 if dim1 and not dim2:
113 if dim1 and dim2:
116 dim2val = self.eval(dim2)
169 var, dim1, dim2 = target
170 if not dim1 and not dim2:
172 elif dim1 and not dim2:
182 elif dim1 and dim2:
184 dim2val = self.eval(dim2)
[all …]
/external/webrtc/webrtc/modules/audio_coding/codecs/ilbc/
Dmy_corr.c30 size_t dim2 /* (i) dimension seq2 */ in WebRtcIlbcfix_MyCorr() argument
47 loops=dim1-dim2+1; in WebRtcIlbcfix_MyCorr()
50 WebRtcSpl_CrossCorrelation(corr, seq2, seq1, dim2, loops, scale, 1); in WebRtcIlbcfix_MyCorr()
Dmy_corr.h33 size_t dim2 /* (i) dimension seq2 */
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensor.h353 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2) in Tensor() argument
354 : m_storage(dim1*dim2, array<Index, 2>(dim1, dim2)) in Tensor()
358 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2, Index dim3) in Tensor() argument
359 : m_storage(dim1*dim2*dim3, array<Index, 3>(dim1, dim2, dim3)) in Tensor()
363 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2, Index dim3, Index dim4) in Tensor() argument
364 : m_storage(dim1*dim2*dim3*dim4, array<Index, 4>(dim1, dim2, dim3, dim4)) in Tensor()
368 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2, Index dim3, Index dim4, Index… in Tensor() argument
369 : m_storage(dim1*dim2*dim3*dim4*dim5, array<Index, 5>(dim1, dim2, dim3, dim4, dim5)) in Tensor()
DTensorMap.h78 …TensorMap(PointerArgType dataPtr, Index dim1, Index dim2) : m_data(dataPtr), m_dimensions(dim1, di… in TensorMap() argument
82 …Map(PointerArgType dataPtr, Index dim1, Index dim2, Index dim3) : m_data(dataPtr), m_dimensions(di… in TensorMap() argument
86 …interArgType dataPtr, Index dim1, Index dim2, Index dim3, Index dim4) : m_data(dataPtr), m_dimensi… in TensorMap() argument
90 …rgType dataPtr, Index dim1, Index dim2, Index dim3, Index dim4, Index dim5) : m_data(dataPtr), m_d… in TensorMap() argument
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h488 int64 dim2 = 0; in CreateR3Projected() local
490 array(dim0, dim1, dim2) = value; in CreateR3Projected()
491 ++dim2; in CreateR3Projected()
493 CHECK_EQ(dim2_size, dim2); in CreateR3Projected()
513 int64 dim2 = 0; in CreateR4Projected() local
517 array(dim0, dim1, dim2, dim3) = value; in CreateR4Projected()
521 ++dim2; in CreateR4Projected()
523 CHECK_EQ(dim2_size, dim2); in CreateR4Projected()
Dreference_util.cc521 WindowDimension dim2; in ConvArray4DGeneralDimensionsDilated() local
522 dim2.set_size( in ConvArray4DGeneralDimensionsDilated()
524 dim2.set_stride(kernel_stride.second); in ConvArray4DGeneralDimensionsDilated()
525 dim2.set_padding_low(paddings[1].first); in ConvArray4DGeneralDimensionsDilated()
526 dim2.set_padding_high(paddings[1].second); in ConvArray4DGeneralDimensionsDilated()
527 dim2.set_window_dilation(rhs_dilation.second); in ConvArray4DGeneralDimensionsDilated()
528 dim2.set_base_dilation(lhs_dilation.second); in ConvArray4DGeneralDimensionsDilated()
529 *window.add_dimensions() = dim2; in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/
Dreduce_slice_ops.cc64 Index dim2 = output.dimension(1); \
66 Index size = dim1 * dim2 * dim3; \
76 XYZ xyz = global_index_to_xyz(global, XYZ(dim1, dim2, dim3)); \
90 thread_pool->ParallelFor(size, std::max(bound / dim2, (Index)1), work); \
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py2027 dim2 = self.size[1] * input_shape[
2032 [input_shape[0], input_shape[1], dim1, dim2, dim3])
2036 dim2 = self.size[1] * input_shape[
2041 [input_shape[0], dim1, dim2, dim3, input_shape[4]])
2279 dim2 = input_shape[3] + 2 * self.padding[1][0]
2281 dim2 = None
2287 [input_shape[0], input_shape[1], dim1, dim2, dim3])
2294 dim2 = input_shape[2] + 2 * self.padding[1][1]
2296 dim2 = None
2302 [input_shape[0], dim1, dim2, dim3, input_shape[4]])
[all …]
/external/tensorflow/tensorflow/core/util/
Dmkl_util_test.cc76 memory::dims dim2 = {3, 4}; in TEST() local
78 b.SetUsrMem(dim2, strides2); in TEST()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc126 const int64 dim2 = InferenceContext::Value(d2); in Merge() local
128 if (dim1 >= 0 && dim2 >= 0) { in Merge()
129 CHECK_EQ(dim1, dim2); in Merge()
131 } else if (dim1 >= 0 && dim2 < 0) { in Merge()
133 } else if (dim1 < 0 && dim2 >= 0) { in Merge()
134 return RefineDim(dim2, result); in Merge()
137 } else if (dim2 < -1) { in Merge()
138 return RefineDim(dim2, result); in Merge()
140 CHECK_EQ(dim1, dim2); in Merge()
/external/swiftshader/src/OpenGL/compiler/
DOutputASM.h318 static int dim2(TIntermNode *m);
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_impl.py2728 for dim1, dim2 in zip(reversed(shape1.dims[:-3]),
2730 if not (dim1 == 1 or dim2 == 1 or dim1.is_compatible_with(dim2)):
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc771 DimensionOrConstant dim2) { in Matrix() argument
772 return MakeShape({dim1, dim2}); in Matrix()

12