Ejemplos de _HandleNestedIndexedSlices en Python

Ejemplo n.º 1

def divide_by_n_tensors_or_indexed_slices(value, n):
    if isinstance(value, ops.IndexedSlices):
        value = gradients_impl._HandleNestedIndexedSlices(value)  # pylint: disable=protected-access
        return ops.IndexedSlices(value.values / n, value.indices,
                                 value.dense_shape)
    else:
        return value / n

Ejemplo n.º 2

Archivo: cross_tower_utils.py Proyecto: abhinav-upadhyay/tensorflow

def divide_by_n_tensors_or_indexed_slices(value, n):
  if isinstance(value, ops.IndexedSlices):
    value = gradients_impl._HandleNestedIndexedSlices(value)  # pylint: disable=protected-access
    return ops.IndexedSlices(
        value.values / n, value.indices, value.dense_shape)
  else:
    return value / n