Searched defs:window_dims (Results 1 – 6 of 6) sorted by relevance
79 std::vector<int64_t> window_dims(input_shape.dims(), 1); in Compile() local
115 const auto& window_dims = window.dimensions(); in AbslHashValue() local
1801 std::vector<int64_t> window_dims(num_spatial_dims); in InferConvolveShape() local
6416 const auto& window_dims = convolution->window().dimensions(); in SwapConvOperands() local
38 std::vector<int64_t> window_dims(num_spatial_dims); in AvgPoolDivideByCountWithGeneralPadding() local
2537 llvm::ArrayRef<int64_t> input_dims, ArrayAttr window_dims, in GetReduceWindowPaddingAsArray()2563 llvm::ArrayRef<int64_t> input_dims, ArrayAttr window_dims, in GetReduceWindowPaddingAsAttr()4515 int64_t window_dims = tensor_rank - num_index_dims; in matchAndRewrite() local6313 SmallVector<int64_t, 4> window_dims(rank, 1); in matchAndRewrite() local