Searched defs:PadTestFloat (Results 1 – 1 of 1) sorted by relevance
70 class PadTestFloat : public PadTest, class73 PadTestFloat() { set_use_bfloat16(GetParam()); } in PadTestFloat() function in xla::__anon8576b9330111::PadTestFloat85 XLA_TEST_P(PadTestFloat, Pad1DS0ToS0Array) { in XLA_TEST_P() argument100 XLA_TEST_P(PadTestFloat, Pad1DS0ToS5Array) { in XLA_TEST_P() argument115 XLA_TEST_P(PadTestFloat, Pad1DS3Array) { in XLA_TEST_P() argument130 XLA_TEST_P(PadTestFloat, Pad4D_2x0x3x2_FloatArray) { in XLA_TEST_P() argument139 TEST_P(PadTestFloat, Pad4DFloat_1x1x3x2_Array) { in TEST_P() argument163 TEST_P(PadTestFloat, Pad4DFloatArrayWithInteriorPadding) { in TEST_P() argument183 TEST_P(PadTestFloat, Pad4DFloatArrayMinorFirstSmall) { in TEST_P() argument225 XLA_TEST_P(PadTestFloat, Pad4DFloatArrayMinorFirstNonTrivialMinorDimensions) { in XLA_TEST_P() argument[all …]