Searched refs:_weight_tensor (Results 1 – 4 of 4) sorted by relevance
278 weight_tensor = head_lib._weight_tensor(features, head._weight_column)280 weight_tensor = head_lib._weight_tensor(
595 self._weight_tensor = weight_tensor607 return self._weight_tensor
1010 self._weight_tensor = weight_tensor1040 return self._weight_tensor
656 weight_tensor = _weight_tensor(features, weight_column_name)1779 def _weight_tensor(features, weight_column_name): function