Searched defs:kernel_sizes (Results 1 – 6 of 6) sorted by relevance
255 std::vector<int64> kernel_sizes(filter->shape().dimensions().begin(), in HandleBatchGroupCount() local604 std::vector<int64> kernel_sizes(filter->shape().dimensions().begin(), in HandleConvolution() local
2897 std::vector<int64> kernel_sizes(kernel_new->shape().dimensions().begin(), in PropagateOnBackpropFilterConv() local
70 kernel_sizes, argument
1119 std::vector<int32> kernel_sizes; in AvgPoolShape() local1560 std::vector<int32> kernel_sizes; in MaxPoolShapeImpl() local1659 std::vector<int32> kernel_sizes; in MaxPoolV2Shape() local1760 std::vector<int32> kernel_sizes; in Pool3DShape() local
497 Eigen::array<TensorIndex, 2> kernel_sizes; in convolution() local
982 def test_local_conv_1d_and_2d(self, input_shape, kernel_sizes, strides, argument