HomeSort by relevance Sort by last modified time
    Searched defs:activation_size (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
conv_depthwise_backprop_filter_test.cc 58 int64 activation_size = option[1]; local
67 config.activation_dims = {batch, activation_size, activation_size, feature};
71 int64 output_space_size = 3 + activation_size - kernel_size;
83 if (activation_size % 2 == 0 && activation_size == kernel_size) {
88 config.activation_dims = {batch, activation_size, activation_size,
conv_depthwise_test.cc 59 int64 activation_size = option[1]; local
68 config.activation_dims = {batch, activation_size, activation_size, feature};
74 if (activation_size == 1 && kernel_size == 2) {
76 config.output_dims = {batch, activation_size + kernel_size - 1,
77 activation_size + kernel_size, feature};
80 config.stride = activation_size - 1;
84 activation_size - kernel_size + 1, feature};
87 config.output_dims = {batch, activation_size - kernel_size + 1,
88 activation_size - kernel_size + 1, feature}
    [all...]
grouped_convolution_test.cc 63 int64 activation_size = option[1]; local
76 config.activation_dims = {batch, activation_size, activation_size,
83 if (activation_size == 1 && kernel_size == 2) {
85 config.output_dims = {batch, activation_size + kernel_size - 1,
86 activation_size + kernel_size, output_feature};
89 config.stride = activation_size - 1;
93 activation_size - kernel_size + 1, output_feature};
96 config.output_dims = {batch, activation_size - kernel_size + 1,
97 activation_size - kernel_size + 1, output_feature}
    [all...]

Completed in 141 milliseconds