| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2_cpu_test.py | 64 self.feature_config = (
107 feature_config=self.feature_config,
123 feature_config=self.feature_config)
125 for feature, config in zip(nest.flatten(features), self.feature_config):
141 feature_config=self.feature_config)
165 feature_config=self.feature_config)
167 for feature, config in zip(nest.flatten(features), self.feature_config):
188 feature_config=self.feature_config)
192 self.feature_config):
217 feature_config=self.feature_config)
[all …]
|
| D | tpu_embedding_v2_test.py | 176 feature_config = (tpu_embedding_v2_utils.FeatureConfig(
181 feature_config, optimizer)
230 feature_config = (tpu_embedding_v2_utils.FeatureConfig(
234 feature_config, optimizer)
359 self.feature_config = (
426 self.feature_config,
1039 feature_config=self.feature_config,
1152 feature_config={
1228 feature_config=self.feature_config,
1268 feature_config=self.feature_config,
[all …]
|
| D | tpu_embedding_v2_utils_test.py | 81 feature_config = tpu_embedding_v2_utils.FeatureConfig(
85 repr(feature_config),
|
| D | tpu_embedding_v2_correctness_test.py | 86 self.feature_config = (
215 feature_config=self.feature_config,
475 feature_config = (
490 feature_config=feature_config,
|
| D | tpu_embedding_v2.py | 246 …feature_config: Union[tpu_embedding_v2_utils.FeatureConfig, Iterable], # pylint:disable=g-bare-ge…
284 self._feature_config = feature_config
305 for feature in nest.flatten(feature_config):
1506 def cpu_embedding_lookup(inputs, weights, tables, feature_config): argument
1571 nest.assert_same_structure(inputs, feature_config)
1578 flat_features = nest.flatten_with_joined_string_paths(feature_config)
1641 return nest.pack_sequence_as(feature_config, outputs)
|
| D | tpu_embedding.py | 2811 for feature, feature_config in six.iteritems(feature_to_config_dict):
2812 if feature_config.table_id in table_to_features_dict_tmp:
2813 table_to_features_dict_tmp[feature_config.table_id].append(feature)
2815 table_to_features_dict_tmp[feature_config.table_id] = [feature]
2816 table_to_num_features_dict_tmp[feature_config.table_id] = 0
2817 if feature_config.max_sequence_length == 0:
2818 table_to_num_features_dict_tmp[feature_config.table_id] = (
2819 table_to_num_features_dict_tmp[feature_config.table_id] + 1)
2821 table_to_num_features_dict_tmp[feature_config.table_id] = (
2822 table_to_num_features_dict_tmp[feature_config.table_id] +
[all …]
|
| /external/tensorflow/tensorflow/python/util/ |
| D | example_parser_configuration.py | 102 feature_config = config.feature_map[key]
105 fixed_config = feature_config.fixed_len_feature
122 feature_config = config.feature_map[key]
123 var_len_feature = feature_config.var_len_feature
172 feature_config = config.feature_map[key]
175 fixed_config = feature_config.fixed_len_feature
192 feature_config = config.feature_map[key]
193 var_len_feature = feature_config.var_len_feature
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.cc | 229 const FixedLenFeature& feature_config = fixed_len_features[d]; in SingleExampleProtoToTensors() local
230 const string& key = feature_config.key; in SingleExampleProtoToTensors()
231 const DataType& dtype = feature_config.dtype; in SingleExampleProtoToTensors()
232 const TensorShape& shape = feature_config.shape; in SingleExampleProtoToTensors()
233 const Tensor& default_value = feature_config.default_value; in SingleExampleProtoToTensors()
271 const VarLenFeature& feature_config = var_len_features[d]; in SingleExampleProtoToTensors() local
272 const string& key = feature_config.key; in SingleExampleProtoToTensors()
273 const DataType& dtype = feature_config.dtype; in SingleExampleProtoToTensors()
373 const VarLenFeature& feature_config = var_len_features[d]; in BatchExampleProtoToTensors() local
374 const DataType& dtype = feature_config.dtype; in BatchExampleProtoToTensors()
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.tpu.experimental.embedding.pbtxt | 29 …argspec: "args=[\'inputs\', \'weights\', \'tables\', \'feature_config\'], varargs=None, keywords=N…
|
| D | tensorflow.tpu.experimental.embedding.-t-p-u-embedding.pbtxt | 13 …argspec: "args=[\'self\', \'feature_config\', \'optimizer\', \'pipeline_execution_with_tensor_core…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.tpu.experimental.embedding.pbtxt | 29 …argspec: "args=[\'inputs\', \'weights\', \'tables\', \'feature_config\'], varargs=None, keywords=N…
|
| D | tensorflow.tpu.experimental.embedding.-t-p-u-embedding.pbtxt | 13 …argspec: "args=[\'self\', \'feature_config\', \'optimizer\', \'pipeline_execution_with_tensor_core…
|
| /external/llvm-project/clang-tools-extra/clangd/test/ |
| D | lit.cfg.py | 21 lit.llvm.llvm_config.feature_config([('--targets-built',
|
| /external/llvm-project/lld/test/ |
| D | lit.cfg.py | 65 llvm_config.feature_config(
|
| /external/llvm-project/lldb/test/Shell/ |
| D | lit.cfg.py | 73 llvm_config.feature_config(
|
| /external/llvm-project/debuginfo-tests/ |
| D | lit.cfg.py | 160 llvm_config.feature_config(
|
| /external/llvm-project/clang/test/ |
| D | lit.cfg.py | 185 llvm_config.feature_config(
|
| /external/llvm-project/llvm/test/ |
| D | lit.cfg.py | 338 llvm_config.feature_config(
|
| /external/llvm-project/llvm/utils/lit/lit/llvm/ |
| D | config.py | 178 def feature_config(self, features): member in LLVMConfig
|