| 70 | }; |
| 71 | |
| 72 | XLA_TEST_P(SelectAndScatterTest, ParamTest) { |
| 73 | auto operand_shape = GetParam().operand_shape; |
| 74 | Array<float> o(operand_shape); |
| 75 | o.FillRandom(1.5f); |
| 76 | auto operand = ConstantFromArray(&builder_, o); |
| 77 | |
| 78 | auto source_shape = GetParam().source_shape; |
| 79 | Array<float> s(source_shape); |
| 80 | s.FillRandom(12.0f); |
| 81 | auto source = ConstantFromArray(&builder_, s); |
| 82 | |
| 83 | SelectAndScatter(operand, ge_f32_, GetParam().window_dimensions, |
| 84 | GetParam().window_strides, GetParam().padding_type, source, |
| 85 | ConstantR0<float>(&builder_, 0.0f), add_f32_); |
| 86 | |
| 87 | ComputeAndCompare(&builder_, {}, ErrorSpec(1e-5, 1e-5)); |
| 88 | } |
| 89 | |
| 90 | INSTANTIATE_TEST_CASE_P( |
| 91 | SelectAndScatterTest_Instantiation, SelectAndScatterTest, |
nothing calls this directly
no test coverage detected