Searched refs:flat_value (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/python/util/ |
D | nest.cc | 114 auto* flat_value = PyList_GetItem(flat_values.get(), i); in FlattenDictItems() local 125 PyDict_SetItem(flat_dictionary, flat_key, flat_value); in FlattenDictItems()
|
/external/tensorflow/tensorflow/python/data/ops/ |
D | optional_ops.py | 243 def _from_components(self, flat_value): argument 245 return _OptionalImpl(flat_value[0], self._element_spec)
|
/external/tensorflow/tensorflow/python/distribute/ |
D | distribute_lib.py | 2495 def merge_fn(_, flat_value): argument 2496 return self.batch_reduce_to(reduce_op, [(v, v) for v in flat_value],
|
/external/tensorflow/tensorflow/python/ops/ragged/ |
D | ragged_tensor_test.py | 2160 def test_fn(flat_value, row_splits): argument 2162 rt = rt_spec._from_components([flat_value, row_splits])
|