Searched defs:kernel_sizes (Results 1 – 7 of 7) sorted by relevance
255 std::vector<int64> kernel_sizes(filter->shape().dimensions().begin(), in HandleBatchGroupCount() local603 std::vector<int64> kernel_sizes(filter->shape().dimensions().begin(), in HandleConvolution() local
2505 std::vector<int64> kernel_sizes(kernel_new->shape().dimensions().begin(), in PropagateOnBackpropFilterConv() local
70 kernel_sizes, argument
422 Eigen::array<TensorIndex, 2> kernel_sizes; in convolution() local
1079 std::vector<int32> kernel_sizes; in AvgPoolShape() local1520 std::vector<int32> kernel_sizes; in MaxPoolShapeImpl() local1619 std::vector<int32> kernel_sizes; in MaxPoolV2Shape() local1720 std::vector<int32> kernel_sizes; in Pool3DShape() local
979 def test_local_conv_1d_and_2d(self, input_shape, kernel_sizes, strides, argument
659 std::vector<int32> kernel_sizes; in RegisterNodeWithPaddingAndStrides() local