Home
last modified time | relevance | path

Searched refs:MAX_DIMS (Results 1 – 5 of 5) sorted by relevance

/external/OpenCL-CTS/test_conformance/non_uniform_work_group/
DTestNonUniformWorkGroup.h28 #define MAX_DIMS 3 macro
33 cl_ulong get_global_size[MAX_DIMS];
34 cl_ulong get_global_offset[MAX_DIMS];
35 cl_ulong get_local_size[MAX_DIMS];
36 cl_ulong get_enqueued_local_size[MAX_DIMS];
37 cl_ulong get_global_id[MAX_DIMS];
38 cl_ulong get_local_id[MAX_DIMS];
39 cl_ulong get_group_id[MAX_DIMS];
40 cl_ulong get_num_groups[MAX_DIMS];
86 size_t _globalSize[MAX_DIMS];
[all …]
DTestNonUniformWorkGroup.cpp314 for (i = _dims; i < MAX_DIMS; i++) in TestNonUniformWorkGroup()
319 for (i = 0; i < MAX_DIMS; i++) in TestNonUniformWorkGroup()
333 for (i = 0; i < MAX_DIMS; i++) in TestNonUniformWorkGroup()
353 for (i = _dims; i < MAX_DIMS; i++) in TestNonUniformWorkGroup()
411 for (unsigned short i = 0; i < MAX_DIMS; i++) { in verifyData()
451 size_t nonRemainderGlobalSize[MAX_DIMS]; in calculateExpectedValues()
452 size_t numberOfPossibleRegions[MAX_DIMS]; in calculateExpectedValues()
476 for (cl_ushort dim = 0; dim < MAX_DIMS; ++dim) { in calculateExpectedValues()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h131 (shape.dims() == 1 && shape.num_elements() <= nvinfer1::Dims::MAX_DIMS)); in IsTrtShapeTensorCompatible()
Dconvert_nodes.cc318 const int max_rank = nvinfer1::Dims::MAX_DIMS + (use_implicit_batch ? 1 : 0); in ValidateTensorProperties()
394 const int max_nb_dims = nvinfer1::Dims::MAX_DIMS + 1; in GetTrtBroadcastShape()
4267 for (int i = 1; i < nvinfer1::Dims::MAX_DIMS; ++i) { in GetTensorDimsWithProtoShape()
5358 nvinfer1::Dims::MAX_DIMS + (params->use_implicit_batch ? 1 : 0)) { in ConvertGather()
5361 nvinfer1::Dims::MAX_DIMS + 1); in ConvertGather()
Dconvert_nodes_test.cc592 std::vector<int64>(nvinfer1::Dims::MAX_DIMS + 2, 1), &output), in TEST_F()