Searched refs:input_num_valid_rows (Results 1 – 1 of 1) sorted by relevance
51 const Tensor& input_num_valid_rows = context->input(1); in Compute() local53 context, input_num_valid_rows.NumElements() == 1, in Compute()57 input_num_valid_rows.NumElements())); in Compute()59 const float num_valid_rows_f = input_num_valid_rows.flat<float>()(0); in Compute()