Searched defs:kernel_sizes (Results 1 – 7 of 7) 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
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
66 kernel_sizes, argument
497 Eigen::array<TensorIndex, 2> kernel_sizes; in convolution() local
473 const IntArrayRef kernel_sizes = kernel.sizes(); in _convolution_depthwise3x3_winograd() local
610 const auto kernel_sizes = kernel_sizes_ref.toDimVector(); in computeFlops() local