Searched defs:input_subset (Results 1 – 2 of 2) sorted by relevance
185 NodeSubset& input_subset = (*node_subsets_)[input_epoch]; in UpdateNode() local
129 Tensor input_subset = input.index_select(0, row_indices); in forward() local