Searched defs:activation_size (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients_test.py | 54 def __init__(self, activation_size, num_layers): argument
67 def fully_connected_model_fn(batch_size, activation_size, num_layers): argument
104 def create_fc_batch_jacobian(batch_size, activation_size, num_layers): argument
162 def create_fc_per_eg_grad(batch_size, activation_size, num_layers): argument
333 def create_fc_per_eg_jacobians(batch_size, activation_size, num_layers): argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conv_depthwise_test.cc | 59 int64 activation_size = option[1]; in GetConv2DTestCases() local
|
| D | conv_depthwise_backprop_filter_test.cc | 58 int64 activation_size = option[1]; in GetConv2DTestCases() local
|
| D | grouped_convolution_test.cc | 63 int64 activation_size = option[1]; in GetConv2DTestCases() local
|