Searched refs:set_use_bfloat16 (Results 1 – 5 of 5) sorted by relevance
77 FloatReverseTest() { set_use_bfloat16(GetParam().use_bfloat16); } in FloatReverseTest()
70 ReduceWindowTest() : builder_(TestName()) { set_use_bfloat16(GetParam()); } in ReduceWindowTest()600 R4ReduceWindowTest() { set_use_bfloat16(::testing::get<1>(GetParam())); } in R4ReduceWindowTest()1007 R3ReduceWindowTest() { set_use_bfloat16(::testing::get<1>(GetParam())); } in R3ReduceWindowTest()1155 R2ReduceWindowTest() { set_use_bfloat16(::testing::get<1>(GetParam())); } in R2ReduceWindowTest()1346 R1ReduceWindowTest() { set_use_bfloat16(::testing::get<1>(GetParam())); } in R1ReduceWindowTest()
75 PadTestFloat() { set_use_bfloat16(GetParam()); } in PadTestFloat()
377 void set_use_bfloat16(bool value) { use_bfloat16_ = value; } in set_use_bfloat16() function
48 ReshapeTest() { set_use_bfloat16(GetParam()); } in ReshapeTest()