Searched refs:is_any_of (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream.h | 90 constexpr bool is_any_of() { in is_any_of() function
95 constexpr bool is_any_of() { in is_any_of() function
96 return std::is_same_v<T, First> || is_any_of<T, Rest...>(); in is_any_of()
785 detail::is_any_of<InputType, Eigen::half, Eigen::bfloat16, float, in ThenBlasGemm()
790 detail::is_any_of<ConstantType, float, Eigen::half>(), in ThenBlasGemm()
794 detail::is_any_of<InputType, Eigen::half, ConstantType>(), in ThenBlasGemm()
1040 detail::is_any_of<InputType, float, Eigen::half, Eigen::bfloat16, in ThenBlasGemmStridedBatched()
1045 (detail::is_any_of<InputType, Eigen::half, Eigen::bfloat16>() && in ThenBlasGemmStridedBatched()
1492 detail::is_any_of<ABType, Eigen::half, Eigen::bfloat16, float, double, in CheckTypesForExtendedBlas()
1504 detail::is_any_of<CType, Eigen::half, Eigen::bfloat16>()), in CheckTypesForExtendedBlas()
[all …]
|