Searched refs:tensor_float_32_execution_enabled (Results 1 – 25 of 28) sorted by relevance
12
112 self.tf32_keep_ = config.tensor_float_32_execution_enabled()150 self.tf32_keep_ = config.tensor_float_32_execution_enabled()193 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
204 self.tf32_keep_ = config.tensor_float_32_execution_enabled()232 self.tf32_keep_ = config.tensor_float_32_execution_enabled()257 self.tf32_keep_ = config.tensor_float_32_execution_enabled()285 self.tf32_keep_ = config.tensor_float_32_execution_enabled()311 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
39 self.tf32_keep_ = config.tensor_float_32_execution_enabled()152 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
42 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
40 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
69 self.tf32_keep_ = config.tensor_float_32_execution_enabled()437 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
43 self.tf32_keep_ = config.tensor_float_32_execution_enabled()301 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
87 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
39 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
65 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
366 self.tf32_keep_ = config.tensor_float_32_execution_enabled()710 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
23 bool tensor_float_32_execution_enabled();
29 bool tensor_float_32_execution_enabled() { return tensor_float_32_enabled; } in tensor_float_32_execution_enabled() function
21 m.def("is_enabled", &tensorflow::tensor_float_32_execution_enabled); in PYBIND11_MODULE()
878 self.assertTrue(config.tensor_float_32_execution_enabled())881 self.assertFalse(config.tensor_float_32_execution_enabled())884 self.assertTrue(config.tensor_float_32_execution_enabled())895 self.assertTrue(config.tensor_float_32_execution_enabled())907 self.assertTrue(config.tensor_float_32_execution_enabled())909 self.assertFalse(config.tensor_float_32_execution_enabled())
28 def tensor_float_32_execution_enabled(): function
845 output->scalar<bool>()() = tensor_float_32_execution_enabled(); in Compute()
2289 allowed = config.tensor_float_32_execution_enabled()2331 if config.tensor_float_32_execution_enabled() and a.dtype == "float32":2336 elif config.tensor_float_32_execution_enabled() and a.dtype == "complex64":
96 name: "tensor_float_32_execution_enabled"
100 name: "tensor_float_32_execution_enabled"
96 return tensor_float_32_execution_enabled() ? ComputationType::kTF32AsF32 in GetBlasComputationType()
391 tensorflow::tensor_float_32_execution_enabled()) { in DoBlasInternalImpl()628 if (tensorflow::tensor_float_32_execution_enabled()) { in DoBlasInternalImpl()953 } else if (!tensorflow::tensor_float_32_execution_enabled()) { in DoBlasInternalImpl()1271 algo = tensorflow::tensor_float_32_execution_enabled() in DoBlasInternalImpl()
441 tensorflow::tensor_float_32_execution_enabled