Searched refs:d_sorted_conv_layer_indexes (Results 1 – 1 of 1) sorted by relevance
187 Tensor* d_sorted_conv_layer_indexes, Tensor* d_sorted_scores, in AllocateGenerationTempTensors() argument204 d_sorted_conv_layer_indexes)); in AllocateGenerationTempTensors()205 TF_RETURN_IF_ERROR(ResetTensor<int32>(d_sorted_conv_layer_indexes, d)); in AllocateGenerationTempTensors()368 Tensor d_sorted_conv_layer_indexes; // output of cub sorting, indices of in Compute() local381 &d_sorted_conv_layer_indexes, &dev_sorted_scores, &dev_boxes, in Compute()402 d_sorted_conv_layer_indexes.flat<int>().data(), in Compute()418 d_sorted_conv_layer_indexes.flat<int>().data(), in Compute()