Searched refs:num_input_rows (Results 1 – 1 of 1) sorted by relevance
48 const int num_input_rows = input_distance_mat.dim_size(0); in Compute() local60 int num_valid_rows = num_input_rows; in Compute()66 context, num_input_rows >= num_valid_rows, in Compute()69 num_input_rows, " rows.")); in Compute()81 context->allocate_output(0, TensorShape({num_input_rows}), in Compute()90 {num_input_rows, num_input_columns}); in Compute()102 std::vector<int> row_to_col_match_vec(num_input_rows, -1); in Compute()122 TTypes<int>::Vec(row_to_col_match_vec.data(), num_input_rows); in Compute()