Searched refs:enable_tensor_float_32_execution (Results 1 – 25 of 34) sorted by relevance
12
109 config.enable_tensor_float_32_execution(self.tf32_keep_)113 config.enable_tensor_float_32_execution(False)147 config.enable_tensor_float_32_execution(self.tf32_keep_)151 config.enable_tensor_float_32_execution(False)190 config.enable_tensor_float_32_execution(self.tf32_keep_)194 config.enable_tensor_float_32_execution(False)
201 config.enable_tensor_float_32_execution(self.tf32_keep_)205 config.enable_tensor_float_32_execution(False)229 config.enable_tensor_float_32_execution(self.tf32_keep_)233 config.enable_tensor_float_32_execution(False)254 config.enable_tensor_float_32_execution(self.tf32_keep_)258 config.enable_tensor_float_32_execution(False)282 config.enable_tensor_float_32_execution(self.tf32_keep_)286 config.enable_tensor_float_32_execution(False)308 config.enable_tensor_float_32_execution(self.tf32_keep_)312 config.enable_tensor_float_32_execution(False)
36 config.enable_tensor_float_32_execution(self.tf32_keep_)40 config.enable_tensor_float_32_execution(False)149 config.enable_tensor_float_32_execution(self.tf32_keep_)153 config.enable_tensor_float_32_execution(False)
39 config.enable_tensor_float_32_execution(self.tf32_keep_)43 config.enable_tensor_float_32_execution(False)
37 config.enable_tensor_float_32_execution(self.tf32_keep_)41 config.enable_tensor_float_32_execution(False)
66 config.enable_tensor_float_32_execution(self.tf32_keep_)70 config.enable_tensor_float_32_execution(False)434 config.enable_tensor_float_32_execution(self.tf32_keep_)438 config.enable_tensor_float_32_execution(False)
40 config.enable_tensor_float_32_execution(self.tf32_keep_)44 config.enable_tensor_float_32_execution(False)298 config.enable_tensor_float_32_execution(self.tf32_keep_)302 config.enable_tensor_float_32_execution(False)
84 config.enable_tensor_float_32_execution(self.tf32_keep_)88 config.enable_tensor_float_32_execution(False)
36 config.enable_tensor_float_32_execution(self.tf32_keep_)40 config.enable_tensor_float_32_execution(False)
62 config.enable_tensor_float_32_execution(self.tf32_keep_)66 config.enable_tensor_float_32_execution(False)
129 config.enable_tensor_float_32_execution(False)
363 config.enable_tensor_float_32_execution(self.tf32_keep_)367 config.enable_tensor_float_32_execution(False)707 config.enable_tensor_float_32_execution(self.tf32_keep_)711 config.enable_tensor_float_32_execution(False)
40 tensorflow::enable_tensor_float_32_execution(false); in XLA_TEST_F()83 tensorflow::enable_tensor_float_32_execution(false); in XLA_TEST_F()110 tensorflow::enable_tensor_float_32_execution(false); in XLA_TEST_F()139 tensorflow::enable_tensor_float_32_execution(false); in XLA_TEST_F()
21 void enable_tensor_float_32_execution(bool enabled);
25 void enable_tensor_float_32_execution(bool enabled) { in enable_tensor_float_32_execution() function
20 m.def("enable", &tensorflow::enable_tensor_float_32_execution); in PYBIND11_MODULE()
224 tensorflow::enable_tensor_float_32_execution(false); in XLA_TEST_P()263 tensorflow::enable_tensor_float_32_execution(false); in XLA_TEST_P()
67 enable_tensor_float_32_execution(false); in SetUp()
28 name: "enable_tensor_float_32_execution"
32 name: "enable_tensor_float_32_execution"
107 enable_tensor_float_32_execution(false); in SetUp()
41 def enable_tensor_float_32_execution(enabled): function
875 config.enable_tensor_float_32_execution(True)880 config.enable_tensor_float_32_execution(False)883 config.enable_tensor_float_32_execution(True)908 config.enable_tensor_float_32_execution(False)