Home
last modified time | relevance | path

Searched defs:dim3 (Results 1 – 25 of 27) sorted by relevance

12

/external/clang/test/CodeGenCUDA/Inputs/
Dcuda.h12 struct dim3 { struct
14 __host__ __device__ dim3(unsigned x, unsigned y = 1, unsigned z = 1) : x(x), y(y), z(z) {} in x() argument
/external/clang/test/SemaCUDA/Inputs/
Dcuda.h15 struct dim3 { struct
17 __host__ __device__ dim3(unsigned x, unsigned y = 1, unsigned z = 1) : x(x), y(y), z(z) {} in x() argument
/external/clang/test/PCH/Inputs/
Dcuda.h12 struct dim3 { struct
14 __host__ __device__ dim3(unsigned x, unsigned y = 1, unsigned z = 1) : x(x), y(y), z(z) {} in x() argument
/external/clang/lib/Headers/
D__clang_cuda_runtime_wrapper.h289 __device__ inline __cuda_builtin_blockDim_t::operator dim3() const { in dim3() function
293 __device__ inline __cuda_builtin_gridDim_t::operator dim3() const { in dim3() function
308 #define dim3 __cuda_builtin_blockDim_t macro
/external/aac/libFDK/include/
DFDK_matrixCalloc.h212 #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.h102 …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.h358 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_reverse_sycl.cpp27 IndexType dim3 = 5; in test_simple_reverse() local
118 IndexType dim3 = 5; in test_expr_reverse() local
Dcxx11_tensor_ifft.cpp100 static void test_sub_fft_ifft_invariant(int dim0, int dim1, int dim2, int dim3) { in test_sub_fft_ifft_invariant()
Dcxx11_tensor_morphing_sycl.cpp36 typename Tensor<DataType, 2 ,DataLayout, IndexType>::Dimensions dim3(6,7); in test_simple_reshape() local
94 typename Tensor<DataType, 5, DataLayout, IndexType>::Dimensions dim3(2,3,1,7,1); in test_reshape_as_lvalue() local
Dcxx11_tensor_index_list.cpp273 int dim3 = 0; in test_dynamic_index_list() local
Dcxx11_tensor_morphing.cpp30 Tensor<float, 2>::Dimensions dim3(2,21); in test_simple_reshape() local
/external/aac/libFDK/src/
DFDK_matrixCalloc.cpp236 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()
/external/armnn/src/backends/reference/workloads/
DSlice.cpp54 unsigned int dim3 = paddedInput[3]; in Slice() local
DStridedSlice.cpp138 int dim3 = armnn::numeric_cast<int>(inputShape[3]); in StridedSlice() local
/external/ComputeLibrary/tests/validation/reference/
DFuseBatchNormalization.cpp84 const unsigned int dim3 = w.shape()[3]; in fuse_batch_normalization_conv_layer() local
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h474 int64_t dim3 = 0; in CreateR4Projected() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
Dkernel_ops_pattern.cc428 auto make_kernel_dim3 = [&](const auto& dim3) { in Rewrite()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc177 void DescribeTensor4D(int dim0, int dim1, int dim2, int dim3, in DescribeTensor4D()
189 void DescribeTensor5D(int dim0, int dim1, int dim2, int dim3, int dim4, in DescribeTensor5D()
/external/pytorch/test/cpp/tensorexpr/
Dtest_kernel.cpp2032 auto run_kernel = [&](int dim1, int dim2, int dim3) { in TEST_F()
2107 auto run_kernel = [&](int dim2, int dim3, int dim4, int dim5) { in TEST_F()
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h920 const int dim3 = reinterpret_input_as_3d ? 1 : input0.tensor_shape()[3]; in compute_mm_shape() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1611 auto get_dim3d = [](mlir::gpu::KernelDim3 dim3) { in EmitLaunchFunc()
/external/angle/src/tests/gl_tests/
DGLSLTest.cpp13022 for (int dim3 = 0; dim3 < 4; ++dim3) in testArrayOfArrayOfSamplerDynamicIndex() local
13235 for (int dim3 = 0; dim3 < 4; ++dim3) in TEST_P() local
/external/angle/src/common/spirv/
Dspirv_instruction_builder_autogen.cpp342 spv::Dim dim3, in WriteTypeImage()
Dspirv_instruction_parser_autogen.cpp330 spv::Dim *dim3, in ParseTypeImage()

12