Home
last modified time | relevance | path

Searched refs:is_built_with_rocm (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Ddepthwise_conv_op_base.py439 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
490 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
513 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
768 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
805 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
829 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
889 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
915 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
939 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
991 if test.is_built_with_rocm():
[all …]
Dconv1d_test.py31 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
Dpooling_ops_test.py211 if use_gpu and data_type == dtypes.float64 and test.is_built_with_rocm():
313 if not test.is_built_with_rocm():
910 + [np.float64] if not test.is_built_with_rocm() else []:
927 + [np.float64] if not test.is_built_with_rocm() else []:
960 + [np.float64] if not test.is_built_with_rocm() else []:
1846 if test.is_built_with_rocm():
1933 if test.is_built_with_rocm():
Dconv_ops_3d_test.py57 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
517 tolerance = 5e-3 if test.is_built_with_rocm() else 1e-3
/external/tensorflow/tensorflow/python/platform/
Dtest.py114 def is_built_with_rocm(): function
170 return is_built_with_cuda() or is_built_with_rocm()
Dbuild_info_test.py27 test.is_built_with_rocm())
Dsysconfig_test.py33 self.assertEqual(build_info["is_rocm_build"], test.is_built_with_rocm())
/external/tensorflow/tensorflow/python/kernel_tests/
Dsegment_reduction_ops_deterministic_test.py112 pred=test.is_built_with_rocm,
134 if not test.is_built_with_rocm():
150 if not test.is_built_with_rocm():
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dsegment_reduction_ops_d9m_test.py122 pred=test.is_built_with_rocm,
148 if not test.is_built_with_rocm():
168 if not test.is_built_with_rocm():
Dbanded_triangular_solve_op_test.py142 pred=test.is_built_with_rocm,
181 pred=test.is_built_with_rocm,
Dcwise_ops_unary_test.py443 pred=test.is_built_with_rocm, skip_message="On ROCm this test fails")
453 pred=test.is_built_with_rocm, skip_message="On ROCm this test fails")
459 pred=test.is_built_with_rocm, skip_message="On ROCm this test fails")
469 pred=test.is_built_with_rocm, skip_message="On ROCm this test fails")
489 pred=test.is_built_with_rocm, skip_message="On ROCm this test fails")
512 pred=test.is_built_with_rocm, skip_message="On ROCm this test fails")
/external/tensorflow/tensorflow/python/eager/
Ddef_function_test_cpu_only.py33 if test.is_built_with_rocm() or test_util.is_xla_enabled():
/external/tensorflow/tensorflow/python/profiler/internal/
Drun_metadata_test.py118 pred=test.is_built_with_rocm,
225 pred=test.is_built_with_rocm,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.test.pbtxt44 name: "is_built_with_rocm"
/external/tensorflow/tensorflow/python/grappler/
Dauto_mixed_precision_test.py461 elif test.is_built_with_rocm():
558 tol = 2e-3 if test.is_built_with_rocm else 1e-3
700 atol = 1e-2 if test.is_built_with_rocm() else 1e-3
728 elif test.is_built_with_rocm():
840 atol = 1e-2 if test.is_built_with_rocm() else tol
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.test.pbtxt60 name: "is_built_with_rocm"
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dfft_ops_test.py344 if test.is_built_with_rocm():
426 if test.is_built_with_rocm() and (rank == 3):
588 if test.is_built_with_rocm():
607 if test.is_built_with_rocm():
Dspectral_ops_test.py294 pred=test.is_built_with_rocm,
/external/tensorflow/tensorflow/compiler/tests/
Dcategorical_op_test.py106 pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.")
Dmatrix_triangular_solve_op_test.py98 if xla_test.test.is_built_with_rocm():
/external/tensorflow/tensorflow/python/ops/
Dinit_ops_test.py176 pred=test.is_built_with_rocm,
Dnn_test.py1584 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float64
1595 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float64
1617 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float64
1664 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float64
1675 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float64
1696 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float64
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinalg_grad_test.py243 if not test_lib.is_built_with_rocm():
Dtridiagonal_matmul_op_test.py193 if test.is_built_with_rocm():
/external/tensorflow/tensorflow/python/framework/
Dconfig_test.py497 if test.is_built_with_rocm():

12