Searched defs:window_dims (Results 1 – 5 of 5) sorted by relevance
79 std::vector<int64> window_dims(input_shape.dims(), 1); in Compile() local
38 std::vector<int64> window_dims(num_spatial_dims); in AvgPoolDivideByCountWithGeneralPadding() local
2139 llvm::ArrayRef<int64_t> input_dims, ArrayAttr window_dims, in GetReduceWindowPaddingAsArray()2166 llvm::ArrayRef<int64_t> input_dims, ArrayAttr window_dims, in GetReduceWindowPaddingAsAttr()3938 int64_t window_dims = tensor_rank - num_index_dims; in matchAndRewrite() local5405 SmallVector<int64_t, 4> window_dims(rank, 1); in matchAndRewrite() local
1799 std::vector<int64> window_dims(num_spatial_dims); in InferConvolveShape() local
5286 const auto& window_dims = convolution->window().dimensions(); in SwapConvOperands() local