Searched refs:dev_boxes_keep_flags (Results 1 – 1 of 1) sorted by relevance
188 Tensor* dev_boxes, Tensor* dev_boxes_keep_flags, int num_images, in AllocateGenerationTempTensors() argument216 dev_boxes_keep_flags)); in AllocateGenerationTempTensors()217 TF_RETURN_IF_ERROR(ResetTensor<int8>(dev_boxes_keep_flags, d)); in AllocateGenerationTempTensors()372 Tensor dev_boxes_keep_flags; // bitmask for keeping the boxes or rejecting in Compute() local382 &dev_boxes_keep_flags, num_images, conv_layer_nboxes, in Compute()425 (char*)dev_boxes_keep_flags.flat<int8>().data())); in Compute()451 char* d_boxes_keep_flags = (char*)dev_boxes_keep_flags.flat<int8>().data(); in Compute()