Searched refs:tensor_float_32_execution_enabled (Results 1 – 13 of 13) 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()
873 self.assertTrue(config.tensor_float_32_execution_enabled())876 self.assertFalse(config.tensor_float_32_execution_enabled())879 self.assertTrue(config.tensor_float_32_execution_enabled())890 self.assertTrue(config.tensor_float_32_execution_enabled())902 self.assertTrue(config.tensor_float_32_execution_enabled())904 self.assertFalse(config.tensor_float_32_execution_enabled())
32 def tensor_float_32_execution_enabled(): function
737 output->scalar<bool>()() = tensor_float_32_execution_enabled(); in Compute()
2059 allowed = config.tensor_float_32_execution_enabled()2101 if config.tensor_float_32_execution_enabled() and a.dtype == "float32":2106 elif config.tensor_float_32_execution_enabled() and a.dtype == "complex64":
96 name: "tensor_float_32_execution_enabled"
400 tensorflow::tensor_float_32_execution_enabled
997 allowed = config.tensor_float_32_execution_enabled()
551 tensorflow::tensor_float_32_execution_enabled()) { in DoBlasInternalImpl()1763 if (tensorflow::tensor_float_32_execution_enabled()) { in DoBlasInternalImpl()2085 } else if (!tensorflow::tensor_float_32_execution_enabled()) { in DoBlasInternalImpl()2392 algo = tensorflow::tensor_float_32_execution_enabled() in DoBlasInternalImpl()
849 if (!tensorflow::tensor_float_32_execution_enabled()) { in IsTensorMathEnabled()1218 allow_tensor_ops = tensorflow::tensor_float_32_execution_enabled(); in Create()