Searched refs:transformation_cache (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 469 transformation_cache = FeatureTransformationCache(features)
473 tensor = column.get_dense_tensor(transformation_cache,
552 transformation_cache = FeatureTransformationCache(features)
562 transformation_cache=transformation_cache,
750 transformation_cache = FeatureTransformationCache(features)
753 outputs[column] = transformation_cache.get(column, state_manager)
2140 def transform_feature(self, transformation_cache, state_manager): argument
2316 def get_dense_tensor(self, transformation_cache, state_manager): argument
2350 def _create_weighted_sum(column, transformation_cache, state_manager, argument
2356 transformation_cache=transformation_cache,
[all …]
|
| D | sequence_feature_column.py | 131 transformation_cache = fc.FeatureTransformationCache(features)
138 transformation_cache, self._state_manager)
518 def transform_feature(self, transformation_cache, state_manager): argument
532 input_tensor = transformation_cache.get(self.key, state_manager)
542 def get_sequence_dense_tensor(self, transformation_cache, state_manager): argument
551 sp_tensor = transformation_cache.get(self, state_manager)
|
| D | feature_column_v2_test.py | 109 def transform_feature(self, transformation_cache, state_manager): argument
111 return transformation_cache.get('a', state_manager)
117 transformation_cache = fc.FeatureTransformationCache(
121 transformation_cache.get(column, None)
123 transformation_cache.get(column, None)
138 def transform_feature(self, transformation_cache, state_manager): argument
145 transformation_cache = fc.FeatureTransformationCache(
148 self.assertEqual('Output', transformation_cache.get(column, None))
149 self.assertEqual('Output', transformation_cache.get(column, None))
163 def transform_feature(self, transformation_cache, state_manager): argument
[all …]
|