Home
last modified time | relevance | path

Searched refs:flatten_with_joined_string_paths (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_embedding_v2.py681 nest.flatten_with_joined_string_paths(gradients),
1039 for path, input_tensor in nest.flatten_with_joined_string_paths(
1254 flat_features = nest.flatten_with_joined_string_paths(self._feature_config)
1339 nest.flatten_with_joined_string_paths(tensors),
1434 nest.flatten_with_joined_string_paths(self._feature_config),
1455 nest.flatten_with_joined_string_paths(self._feature_config),
H A Dtpu_embedding_for_serving.py285 flat_features = nest.flatten_with_joined_string_paths(feature_config)
H A Dtpu_embedding_v1.py267 flat_features = nest.flatten_with_joined_string_paths(self._feature_config)
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dcomposite_tensor_test.py128 result_with_string_paths = nest.flatten_with_joined_string_paths(
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dnest.py1651 def flatten_with_joined_string_paths(structure, separator="/", function
H A Dnest_test.py1136 nest.flatten_with_joined_string_paths(inputs, separator="/"),
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dtpu_strategy.py978 specs = nest.flatten_with_joined_string_paths(element_spec)