Searched refs:tensor_float_32_execution_enabled (Results 1 – 12 of 12) sorted by relevance
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()
92 name: "tensor_float_32_execution_enabled"
30 def tensor_float_32_execution_enabled(): function
821 self.assertTrue(config.tensor_float_32_execution_enabled())832 self.assertTrue(config.tensor_float_32_execution_enabled())834 self.assertFalse(config.tensor_float_32_execution_enabled())
1952 allowed = config.tensor_float_32_execution_enabled()1994 if config.tensor_float_32_execution_enabled() and a.dtype == "float32":1999 elif config.tensor_float_32_execution_enabled() and a.dtype == "complex64":
380 tensorflow::tensor_float_32_execution_enabled
1017 allowed = config.tensor_float_32_execution_enabled()
546 tensorflow::tensor_float_32_execution_enabled()) { in DoBlasInternalImpl()2107 } else if (!tensorflow::tensor_float_32_execution_enabled()) { in DoBlasInternalImpl()2451 algo = tensorflow::tensor_float_32_execution_enabled() in DoBlasInternalImpl()
767 if (!tensorflow::tensor_float_32_execution_enabled()) { in IsTensorMathEnabled()1116 allow_tensor_ops = tensorflow::tensor_float_32_execution_enabled(); in Create()