Searched refs:MAX_DIMS (Results 1 – 5 of 5) sorted by relevance
28 #define MAX_DIMS 3 macro33 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 …]
314 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()
131 (shape.dims() == 1 && shape.num_elements() <= nvinfer1::Dims::MAX_DIMS)); in IsTrtShapeTensorCompatible()
318 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()
592 std::vector<int64>(nvinfer1::Dims::MAX_DIMS + 2, 1), &output), in TEST_F()