H A D | select_and_scatter_test.cc | 45 tensorflow::gtl::ArraySlice<int64> window_dimensions; member in struct:xla::__anon25784::SelectAndScatterTestParam 84 operand, ge_f32_, GetParam().window_dimensions, GetParam().window_strides, 203 builder_.SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3}, 215 builder_.SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3}, 226 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{3}, 237 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{3}, 249 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 3}, 263 builder_.SelectAndScatter(reshape, ge_s32_, /*window_dimensions=*/{2, 3}, 275 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 3}, 287 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions [all...] |