| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column_v2.py | 545 def get_dense_tensor(self, transformation_cache, state_manager): argument
549 self, transformation_cache, state_manager)
554 self, transformation_cache, state_manager)
558 tensor = transformation_cache.get(self.get_feature_key_name(),
591 def get_sequence_dense_tensor(self, transformation_cache, state_manager): argument
595 self, transformation_cache, state_manager)
600 self, transformation_cache, state_manager)
602 tensor = transformation_cache.get(self.get_feature_key_name(),
604 tensor_lengths = transformation_cache.get(
704 self, transformation_cache, state_manager): argument
[all …]
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 419 transformation_cache = FeatureTransformationCache(features)
424 outputs[column] = transformation_cache.get(column, state_manager)
1947 def transform_feature(self, transformation_cache, state_manager): argument
2129 def get_dense_tensor(self, transformation_cache, state_manager): argument
2163 def _create_weighted_sum(column, transformation_cache, state_manager, argument
2169 transformation_cache=transformation_cache,
2176 transformation_cache=transformation_cache,
2181 def _create_dense_column_weighted_sum(column, transformation_cache, argument
2184 tensor = column.get_dense_tensor(transformation_cache, state_manager)
2207 def get_sparse_tensors(self, transformation_cache, state_manager): argument
[all …]
|
| D | sequence_feature_column.py | 408 def transform_feature(self, transformation_cache, state_manager): argument
422 input_tensor = transformation_cache.get(self.key, state_manager)
432 def get_sequence_dense_tensor(self, transformation_cache, state_manager): argument
441 sp_tensor = transformation_cache.get(self, state_manager)
|
| D | feature_column_v2_test.py | 130 def transform_feature(self, transformation_cache, state_manager): argument
132 return transformation_cache.get('a', state_manager)
138 transformation_cache = fc.FeatureTransformationCache(
142 transformation_cache.get(column, None)
144 transformation_cache.get(column, None)
159 def transform_feature(self, transformation_cache, state_manager): argument
166 transformation_cache = fc.FeatureTransformationCache(
169 self.assertEqual('Output', transformation_cache.get(column, None))
170 self.assertEqual('Output', transformation_cache.get(column, None))
184 def transform_feature(self, transformation_cache, state_manager): argument
[all …]
|
| /external/tensorflow/tensorflow/python/keras/feature_column/ |
| D | dense_features.py | 160 transformation_cache = fc.FeatureTransformationCache(features)
166 transformation_cache, self._state_manager, training=training)
168 tensor = column.get_dense_tensor(transformation_cache,
|
| D | sequence_feature_column.py | 149 transformation_cache = fc.FeatureTransformationCache(features)
157 transformation_cache, self._state_manager, training=training)
160 transformation_cache, self._state_manager)
|
| D | dense_features_v2_test.py | 153 def training_aware_get_dense_tensor(transformation_cache, argument
156 return transformation_cache.get(price2, state_manager, training=training)
158 def training_aware_transform_feature(transformation_cache, argument
161 input_tensor = transformation_cache.get(
|