| D | conv_ops_test.cc | 1105 int filter_w, int filter_h, int out_depth) { in Conv2D() argument
1109 Tensor filter_t = MakeRandomTensor({filter_w, filter_h, in_depth, out_depth}); in Conv2D()
1129 int filter_h, int out_depth) { in Conv2DWithBias() argument
1131 Conv2D(batch, height, width, in_depth, filter_w, filter_h, out_depth); in Conv2DWithBias()
1154 int filter_h, in Conv2DWithBiasAndRelu() argument
1157 batch, height, width, in_depth, filter_w, filter_h, out_depth); in Conv2DWithBiasAndRelu()
1175 int filter_w, int filter_h, in Conv2DWithBatchNorm() argument
1178 Conv2D(batch, height, width, in_depth, filter_w, filter_h, out_depth); in Conv2DWithBatchNorm()
1210 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in Conv2DWithBatchNormAndRelu() argument
1213 batch, height, width, in_depth, filter_w, filter_h, out_depth); in Conv2DWithBatchNormAndRelu()
[all …]
|