Searched refs:window_dims (Results 1 – 3 of 3) sorted by relevance
81 std::vector<int64> window_dims(input_shape.dims(), 1); in Compile() local82 window_dims[axis] = input_shape.dim_size(axis); in Compile()106 window_dims, window_strides, in Compile()
38 std::vector<int64> window_dims(num_spatial_dims); in AvgPoolDivideByCountWithGeneralPadding() local46 window_dims[i] = dim; in AvgPoolDivideByCountWithGeneralPadding()69 return Div(sums, counts, window_dims); in AvgPoolDivideByCountWithGeneralPadding()
1761 std::vector<int64> window_dims(num_spatial_dims); in InferConvolveShape() local1763 window_dims[i] = window.dimensions(i).size(); in InferConvolveShape()1765 if (kernel_spatial_dims != window_dims) { in InferConvolveShape()