Searched refs:window_sizes (Results 1 – 2 of 2) sorted by relevance
173 absl::Span<const int64> operand_dims, absl::Span<const int64> window_sizes, in CheckIndexValidity() argument176 DCHECK_EQ(operand_dims.size(), window_sizes.size()); in CheckIndexValidity()191 max_valid_index[i] = operand_dims[i] - window_sizes[i]; in CheckIndexValidity()215 return MakeBroadcastHlo(valid_index_reduced, {}, window_sizes); in CheckIndexValidity()
184 window_sizes = constant_op.constant([5, 10], dtype=dtypes.int64)185 return window_sizes[key]