HomeSort by relevance Sort by last modified time
    Searched refs:event_dims (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/
batch_normalization_test.py 42 def _reduction_axes(self, input_shape, event_dims):
43 if isinstance(event_dims, int):
44 event_dims = [event_dims]
46 # Convert event_dims to non-negative indexing.
47 event_dims = list(event_dims)
48 for idx, x in enumerate(event_dims):
50 event_dims[idx] = ndims + x
51 return tuple(i for i in range(ndims) if i not in event_dims)
    [all...]
  /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
batch_normalization.py 260 event_dims = self.batchnorm.axis
261 reduction_axes = [i for i in range(len(input_shape)) if i not in event_dims]
272 # `gamma` and `log Var(y)` reductions over event_dims.

Completed in 3217 milliseconds