Searched defs:cub_temp_storage_bytes (Results 1 – 1 of 1) sorted by relevance
189 int conv_layer_nboxes, size_t cub_temp_storage_bytes, in AllocateGenerationTempTensors()375 size_t cub_temp_storage_bytes = in Compute() local