Searched refs:flatten_nested_indexed_slices (Results 1 – 2 of 2) sorted by relevance
594 def flatten_nested_indexed_slices(grad): function600 g = flatten_nested_indexed_slices(grad.values)622 grads = [flatten_nested_indexed_slices(x) for x in grads]
585 value = backprop.flatten_nested_indexed_slices(value)