Searched refs:cub_sort_temp_storage_bytes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 413 size_t cub_sort_temp_storage_bytes = 0; in DoNMS() local
428 nullptr, cub_sort_temp_storage_bytes, in DoNMS()
441 DataType::DT_INT8, TensorShape({(int64)cub_sort_temp_storage_bytes}), in DoNMS()
468 d_cub_sort_buffer.flat<int8>().data(), cub_sort_temp_storage_bytes, in DoNMS()
|
| D | generate_box_proposals_op.cu.cc | 347 size_t cub_sort_temp_storage_bytes = 0; in Compute() local
352 nullptr, cub_sort_temp_storage_bytes, flt_ptr, flt_ptr, int_ptr, in Compute()
376 std::max(cub_sort_temp_storage_bytes, cub_select_temp_storage_bytes); in Compute()
|