Searched refs:cols_to_output_tensors (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/ |
H A D | feature_column.py | 173 cols_to_output_tensors=None, argument 213 if cols_to_output_tensors is not None: 214 cols_to_output_tensors[column] = output_tensor 235 cols_to_output_tensors=None): argument 298 cols_to_output_tensors=cols_to_output_tensors)
|
H A D | feature_column_v2_test.py | 1817 cols_to_output_tensors = {} 1823 features, all_cols, cols_to_output_tensors=cols_to_output_tensors) 1828 self.assertCountEqual(all_cols, cols_to_output_tensors.keys()) 1830 output_tensors = [tensor for tensor in cols_to_output_tensors.values()]
|
H A D | feature_column_test.py | 1657 cols_to_output_tensors = {} 1663 features, all_cols, cols_to_output_tensors=cols_to_output_tensors) 1668 self.assertCountEqual(all_cols, cols_to_output_tensors.keys()) 1670 output_tensors = [tensor for tensor in cols_to_output_tensors.values()]
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
H A D | tensorflow.feature_column.pbtxt | 37 …ns\', \'weight_collections\', \'trainable\', \'cols_to_vars\', \'cols_to_output_tensors\'], vararg…
|