| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| H A D | cwise_ops_unary_test.py | 74 def _compareCpu(self, x, np_func, tf_func, grad_rtol=None, grad_atol=None): argument
127 def _compareSparseCpu(self, x, np_func, tf_func, tol): argument
133 def _compareGpu(self, x, np_func, tf_func): argument
144 def _compareSparseGpu(self, x, np_func, tf_func, tol): argument
150 def _compareBoth(self, x, np_func, tf_func, grad_tol=None): argument
155 def _compareBothSparse(self, x, np_func, tf_func, tol=None): argument
|
| H A D | cwise_ops_binary_test.py | 75 def _compareCpu(self, x, y, np_func, tf_func, also_compare_variables=False): argument
116 tf_func, argument
146 tf_func, argument
172 def _compareGpu(self, x, y, np_func, tf_func): argument
183 def _compareBoth(self, x, y, np_func, tf_func, also_compare_variables=False): argument
438 def _compareBCast(self, xs, ys, dtype, np_func, tf_func): argument
887 def _compare(self, x, y, np_func, tf_func): argument
915 def _compareBCast(self, xs, ys, dtype, np_func, tf_func): argument
924 def _testBCastByFunc(self, np_func, tf_func, include_complex=False): argument
|
| H A D | basic_gpu_test.py | 37 def _compareGPU(self, x, y, np_func, tf_func): argument
89 def _compare(self, x, np_func, tf_func, use_gpu): argument
163 def _compareGpu(self, x, y, np_func, tf_func): argument
|
| H A D | cwise_ops_test.py | 118 def _compare(self, x, y, np_func, tf_func): argument
148 def _compareBCast(self, xs, ys, dtype, np_func, tf_func): argument
157 def _testBCastByFunc(self, np_func, tf_func, include_complex=False): argument
223 def _compareBinary(self, x, y, np_func, tf_func, use_gpu=False): argument
851 def _compareBinary(self, x, y, dtype, np_func, tf_func): argument
866 def _compareUnary(self, x, dtype, np_func, tf_func): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_outside_compilation_test.py | 96 def _rewrite_func_wrapper(tf_func): argument
108 def _tpu_partitioned_call_wrapper(tf_func): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | utils.cc | 32 auto tf_func = module.lookupSymbol<LLVMFuncOp>(func_name); in GetOrInsertLLVMFunction() local
|