Searched defs:kernel_sizes (Results 1 – 5 of 5) sorted by relevance
260 std::vector<int64_t> kernel_sizes(filter->shape().dimensions().begin(), in HandleBatchGroupCount() local609 std::vector<int64_t> kernel_sizes(filter->shape().dimensions().begin(), in HandleConvolution() local
3388 std::vector<int64_t> kernel_sizes(kernel_new->shape().dimensions().begin(), in PropagateOnBackpropFilterConv() local
66 kernel_sizes, argument
497 Eigen::array<TensorIndex, 2> kernel_sizes; in convolution() local
1134 std::vector<int32> kernel_sizes; in AvgPoolShape() local1611 std::vector<int32> kernel_sizes; in MaxPoolShapeImpl() local1710 std::vector<int32> kernel_sizes; in MaxPoolV2Shape() local1811 std::vector<int32> kernel_sizes; in Pool3DShape() local