Searched refs:flatten_nested_indexed_slices (Results 1 – 2 of 2) sorted by relevance
586 def flatten_nested_indexed_slices(grad): function592 g = flatten_nested_indexed_slices(grad.values)615 grads = [flatten_nested_indexed_slices(x) for x in grads]
584 value = backprop.flatten_nested_indexed_slices(value)