Searched refs:window_sizes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 173 absl::Span<const int64> operand_dims, absl::Span<const int64> window_sizes, in CheckIndexValidity() argument
176 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()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 30 absl::Span<const int64> window_sizes, in DynamicStridedSlice() argument
32 XlaOp sliced_input = DynamicSlice(input, base_indices, window_sizes); in DynamicStridedSlice()
35 sliced_input = Slice(sliced_input, std::vector<int64>(window_sizes.size()), in DynamicStridedSlice()
36 window_sizes, strides); in DynamicStridedSlice()
|
| D | slicing.h | 29 absl::Span<const int64> window_sizes,
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | group_by_window_test.py | 188 window_sizes = constant_op.constant([5, 10], dtype=dtypes.int64)
189 return window_sizes[key]
|