Home
last modified time | relevance | path

Searched refs:enable_tensor_float_32_execution (Results 1 – 25 of 34) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A Dlinear_operator_low_rank_update_test.py201 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)
H A Dlinear_operator_tridiag_test.py109 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)
H A Dlinear_operator_composition_test.py36 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)
H A Dlinear_operator_block_diag_test.py66 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)
H A Dlinear_operator_permutation_test.py39 config.enable_tensor_float_32_execution(self.tf32_keep_)
43 config.enable_tensor_float_32_execution(False)
H A Dlinear_operator_householder_test.py37 config.enable_tensor_float_32_execution(self.tf32_keep_)
41 config.enable_tensor_float_32_execution(False)
H A Dlinear_operator_identity_test.py40 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)
H A Dlinear_operator_inversion_test.py37 config.enable_tensor_float_32_execution(self.tf32_keep_)
41 config.enable_tensor_float_32_execution(False)
H A Dlinear_operator_diag_test.py36 config.enable_tensor_float_32_execution(self.tf32_keep_)
40 config.enable_tensor_float_32_execution(False)
H A Dlinear_operator_kronecker_test.py84 config.enable_tensor_float_32_execution(self.tf32_keep_)
88 config.enable_tensor_float_32_execution(False)
H A Dlinear_operator_circulant_test.py363 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)
H A Dlinear_operator_block_lower_triangular_test.py62 config.enable_tensor_float_32_execution(self.tf32_keep_)
66 config.enable_tensor_float_32_execution(False)
H A Dlinear_operator_adjoint_test.py39 config.enable_tensor_float_32_execution(self.tf32_keep_)
43 config.enable_tensor_float_32_execution(False)
H A Dlinear_operator_lower_triangular_test.py129 config.enable_tensor_float_32_execution(False)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dqr_test.cc40 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dcholesky_test.cc224 tensorflow::enable_tensor_float_32_execution(false); in XLA_TEST_P()
263 tensorflow::enable_tensor_float_32_execution(false); in XLA_TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconfig_test.py875 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)
H A Dconfig.py41 def enable_tensor_float_32_execution(enabled): function
/aosp_15_r20/external/tensorflow/tensorflow/core/platform/
H A Dtensor_float_32_utils.h21 void enable_tensor_float_32_execution(bool enabled);
H A Dtensor_float_32_utils.cc25 void enable_tensor_float_32_execution(bool enabled) { in enable_tensor_float_32_execution() function
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dtensor_float_32.cc20 m.def("enable", &tensorflow::enable_tensor_float_32_execution); in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/
H A Darray_grad_test.cc67 enable_tensor_float_32_execution(false); in SetUp()
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.config.experimental.pbtxt28 name: "enable_tensor_float_32_execution"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.config.experimental.pbtxt32 name: "enable_tensor_float_32_execution"
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dgradient_checker_test.cc107 enable_tensor_float_32_execution(false); in SetUp()

12