Home
last modified time | relevance | path

Searched defs:dim2 (Results 1 – 25 of 30) 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
/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.h78 …EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index dim1, Index dim2) : m_data(dataPtr), m… in TensorMap()
82 …EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index dim1, Index dim2, Index dim3) : m_data… in TensorMap()
86 …EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index dim1, Index dim2, Index dim3, Index di… in TensorMap()
90 …EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index dim1, Index dim2, Index dim3, Index di… 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.cpp273 int dim2 = 1; in test_dynamic_index_list() local
304 int dim2 = 1; in test_mixed_index_list() local
368 Eigen::IndexList<Eigen::type2index<1>, int> dim2; in test_dim_check() local
Dcxx11_tensor_morphing.cpp28 Tensor<float, 2>::Dimensions dim2(6,7); in test_simple_reshape() local
/external/webrtc/modules/audio_coding/codecs/ilbc/
Dmy_corr.c30 size_t dim2 /* (i) dimension seq2 */ in WebRtcIlbcfix_MyCorr()
/external/llvm-project/mlir/lib/Dialect/
DTraits.cpp24 auto dim2 = std::get<1>(dimensions); in staticallyKnownBroadcastable() local
171 auto isCompatible = [](int64_t dim1, int64_t dim2) { in areCompatibleShapes()
/external/llvm-project/mlir/lib/IR/
DTypeUtilities.cpp60 int64_t dim2 = std::get<1>(dims); in verifyCompatibleShape() local
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h455 int64 dim2 = 0; in CreateR3Projected() local
480 int64 dim2 = 0; in CreateR4Projected() local
Dreference_util.cc455 WindowDimension dim2; in ConvArray4DGeneralDimensionsDilated() local
/external/tensorflow/tensorflow/core/util/
Dmkl_util_test.cc74 memory::dims dim2 = {3, 4}; in TEST() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_types.cc46 int64_t dim2 = std::get<1>(dims); in GetCastCompatibleShape() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc176 void DescribeTensor4D(int dim0, int dim1, int dim2, int dim3, in DescribeTensor4D()
188 void DescribeTensor5D(int dim0, int dim1, int dim2, int dim3, int dim4, in DescribeTensor5D()
Dgraph_properties.cc133 const int64 dim2 = InferenceContext::Value(d2); in Merge() local
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc713 DimensionOrConstant dim2) { in Matrix()
/external/angle/src/tests/gl_tests/
DGLSLTest.cpp9566 for (int dim2 = 0; dim2 < 3; ++dim2) in TEST_P() local
9675 for (int dim2 = 0; dim2 < 3; ++dim2) in TEST_P() local
9808 for (int dim2 = 0; dim2 < 3; ++dim2) in TEST_P() local
9985 for (int dim2 = 0; dim2 < 3; ++dim2) in TEST_P() local
10115 for (int dim2 = 0; dim2 < 3; ++dim2) in TEST_P() local
/external/javassist/src/main/javassist/compiler/
DCodeGen.java1088 int dim2 = arrayDim; in atStringConcatExpr() local
/external/llvm-project/polly/lib/External/isl/
Disl_space.c3192 isl_size dim1, dim2; in isl_space_cmp_type() local
/external/swiftshader/src/OpenGL/compiler/
DOutputASM.cpp3827 int OutputASM::dim2(TIntermNode *m) in dim2() function in glsl::OutputASM

12