OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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