Searched defs:batch_input_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/normalization/ |
| D | layer_normalization_test.py | 178 def _expected_layer_norm(self, x, beta, gamma, batch_input_shape, axis, argument
190 def _test_forward_pass(self, batch_input_shape, axis, fp64_tol=1e-14, argument
248 def _test_backward_pass(self, batch_input_shape, axis, fp64_tol=1e-5, argument
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops.cc | 1226 TensorShape batch_input_shape({timelen, batch_size, input_size}); in Compute() local
|