Searched refs:num_input_columns (Results 1 – 1 of 1) sorted by relevance
49 const int num_input_columns = input_distance_mat.dim_size(1); in Compute() local75 valid_top_k = num_valid_rows * num_input_columns; in Compute()85 context->allocate_output(1, TensorShape({num_input_columns}), in Compute()90 {num_input_rows, num_input_columns}); in Compute()96 for (int index2 = 0; index2 < num_input_columns; index2++) { in Compute()103 std::vector<int> col_to_row_match_vec(num_input_columns, -1); in Compute()124 TTypes<int>::Vec(col_to_row_match_vec.data(), num_input_columns); in Compute()