Searched defs:FeatureConfig (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_for_serving.py | 87 …feature_config: Union[tpu_embedding_v2_utils.FeatureConfig, Iterable], # pylint:disable=g-bare-ge…
211 …feature_config: Union[tpu_embedding_v2_utils.FeatureConfig, Iterable] # pylint:disable=g-bare-gen…
325 feature: tpu_embedding_v2_utils.FeatureConfig) -> ops.Tensor:
367 feature: tpu_embedding_v2_utils.FeatureConfig) -> ops.Tensor:
|
| D | tpu_embedding_v1.py | 131 …feature_config: Union[tpu_embedding_v2_utils.FeatureConfig, Iterable], # pylint:disable=g-bare-ge…
310 feature: tpu_embedding_v2_utils.FeatureConfig) -> ops.Tensor:
355 feature: tpu_embedding_v2_utils.FeatureConfig) -> ops.Tensor:
|
| D | tpu_embedding_base.py | 36 …feature_config: Union[tpu_embedding_v2_utils.FeatureConfig, Iterable], # pylint:disable=g-bare-ge…
|
| D | tpu_embedding_v2.py | 238 …feature_config: Union[tpu_embedding_v2_utils.FeatureConfig, Iterable], # pylint:disable=g-bare-ge…
940 flat_features: List[tpu_embedding_v2_utils.FeatureConfig],
|
| D | tpu_embedding_v2_utils.py | 959 class FeatureConfig: class
|
| D | tpu_embedding.py | 147 class FeatureConfig( class
|