Searched refs:num_outputs_t (Results 1 – 2 of 2) sorted by relevance
772 Tensor* num_outputs_t = nullptr; in Compute() local774 1, tensorflow::TensorShape{}, &num_outputs_t)); in Compute()775 num_outputs_t->scalar<int32>().setConstant(num_valid_outputs); in Compute()849 Tensor* num_outputs_t = nullptr; in Compute() local851 2, tensorflow::TensorShape{}, &num_outputs_t)); in Compute()852 num_outputs_t->scalar<int32>().setConstant(num_valid_outputs); in Compute()
707 Tensor* num_outputs_t = nullptr; in Compute() local710 &num_outputs_t)); in Compute()717 device.memcpy((num_outputs_t->flat<int>().data()), &num_boxes, in Compute()728 device.memcpyHostToDevice((num_outputs_t->flat<int>().data()), &num_outputs, in Compute()