Searched defs:KeyedJaggedTensor (Results 1 – 1 of 1) sorted by relevance
43 def forward(self, features: "KeyedJaggedTensor") -> "KeyedJaggedTensor":86 def f(id_list_features: KeyedJaggedTensor):