| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| H A D | linear_operator_low_rank_update_test.py | 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()
|
| H A D | linear_operator_tridiag_test.py | 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()
|
| H A D | linear_operator_composition_test.py | 39 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
152 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
|
| H A D | linear_operator_block_diag_test.py | 69 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
437 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
|
| H A D | linear_operator_permutation_test.py | 42 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
|
| H A D | linear_operator_householder_test.py | 40 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
|
| H A D | linear_operator_identity_test.py | 43 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
301 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
|
| H A D | linear_operator_inversion_test.py | 40 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
|
| H A D | linear_operator_diag_test.py | 39 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
|
| H A D | linear_operator_kronecker_test.py | 87 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
|
| H A D | linear_operator_circulant_test.py | 366 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
710 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
|
| H A D | linear_operator_block_lower_triangular_test.py | 65 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
|
| H A D | linear_operator_adjoint_test.py | 42 self.tf32_keep_ = config.tensor_float_32_execution_enabled()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | config_test.py | 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())
|
| H A D | config.py | 28 def tensor_float_32_execution_enabled(): function
|
| H A D | test_ops.cc | 845 output->scalar<bool>()() = tensor_float_32_execution_enabled(); in Compute()
|
| H A D | test_util.py | 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":
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/platform/ |
| H A D | tensor_float_32_utils.h | 23 bool tensor_float_32_execution_enabled();
|
| H A D | tensor_float_32_utils.cc | 29 bool tensor_float_32_execution_enabled() { return tensor_float_32_enabled; } in tensor_float_32_execution_enabled() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/util/ |
| H A D | tensor_float_32.cc | 21 m.def("is_enabled", &tensorflow::tensor_float_32_execution_enabled); in PYBIND11_MODULE()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.config.experimental.pbtxt | 96 name: "tensor_float_32_execution_enabled"
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.config.experimental.pbtxt | 100 name: "tensor_float_32_execution_enabled"
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | matmul_util.cc | 96 return tensor_float_32_execution_enabled() ? ComputationType::kTF32AsF32 in GetBlasComputationType()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_blas.cc | 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/def_file_filter/ |
| H A D | symbols_pybind.txt | 441 tensorflow::tensor_float_32_execution_enabled
|