Searched refs:strict_cast (Results 1 – 9 of 9) sorted by relevance
40 strict_cast(T t);45 strict_cast(T t) { in strict_cast() function50 __device__ __host__ EIGEN_STRONG_INLINE float strict_cast<float, Eigen::half>( in strict_cast() function57 strict_cast<Eigen::half, float>(float t) { in strict_cast() function82 U input = strict_cast<U>(logits[tid]); in GenerateNormalizedProb()83 U max_val = strict_cast<U>(ldg(max_logits + row)); in GenerateNormalizedProb()92 output[tid] = strict_cast<T>(result); in GenerateNormalizedProb()106 strict_cast<U>(logits_[gid] - ldg(max_logits_ + gid / num_cols_)); in operator ()()
67 *max = strict_cast<int64_t>(iter_->first) + 1; in Get()90 IncreaseSumAndCount(strict_cast<int64_t>(count) * value, count); in Accumulate()118 if (strict_cast<int64_t>(min) + 1 != max) in AddSubtractImpl()
72 *max = strict_cast<int64_t>(iter_->first) + 1; in Get()130 IncreaseSumAndCount(strict_cast<int64_t>(count) * value, count); in Accumulate()204 if (strict_cast<int64_t>(min) + 1 != max) in AddSubtractImpl()
60 IncreaseSumAndCount(strict_cast<int64_t>(count) * value, count); in Accumulate()406 *max = strict_cast<int64_t>(bucket_ranges_->range(index_ + 1)); in Get()
240 IncreaseSumAndCount(strict_cast<int64_t>(value) * count, count); in AccumulateSingleSample()
140 constexpr Dst strict_cast(Src value) { in strict_cast() function195 : value_(strict_cast<T>(rhs.value_)) {}201 : value_(strict_cast<T>(value)) {}259 using internal::strict_cast;
44 strict_cast(T t);49 strict_cast(T t) { in strict_cast() function55 strict_cast<Eigen::half, float>(float t) { in strict_cast() function64 t.device(d) = t.constant(strict_cast<T>(0.f)); in operator ()()71 t.device(d) = t.constant(strict_cast<T>(0.f)); in operator ()()94 T forget_bias_t = strict_cast<T>(forget_bias); in lstm_gates()95 T cell_clip_t = strict_cast<T>(cell_clip); in lstm_gates()
209 constexpr Dst strict_cast(Src value) {264 : value_(strict_cast<T>(rhs.value_)) {}270 : value_(strict_cast<T>(value)) {}330 using internal::strict_cast;
60 The `strict_cast` enforces type restrictions at compile-time and results in62 `strict_cast` assignment will fail to compile if the destination type cannot67 int int_value = strict_cast<int>(byte_value);175 * `strict_cast<>()` - Analogous to `static_cast` for numeric types, except199 the `strict_cast` template, and can perform valid arithmetic comparisons