| /external/gemmlowp/meta/ |
| streams.h | 47 template <typename InType>
48 class StreamUtil<InType, RowMajor> {
50 static const InType* Offset(const RowMajor& params, const InType* source,
52 return reinterpret_cast<const InType*>(
54 offset_stride * params.stride + offset_advance * sizeof(InType));
57 static InType* Offset(const RowMajor& params, InType* source,
59 return reinterpret_cast<InType*>(reinterpret_cast<std::uint8_t*>(source) +
61 offset_advance * sizeof(InType));
[all...] |
| base.h | 49 typedef InType_ InType;
64 const InType* lhs;
65 const InType* rhs;
76 template <typename InType, int lanes_count, int pack_size, int leftovers,
80 static void Pack(const InType* in, const StreamParams& params, InType* out);
91 template <typename InType, typename StreamType>
94 static const InType* Offset(const StreamType& params, const InType* source,
100 template <typename InType, typename OutType, typename Kernel
[all...] |
| transform_kernels.h | 70 template <typename InType, typename OutType, int kernel_size, int leftovers>
71 class Transform1DKernel<InType, OutType, Quantize, kernel_size, leftovers> {
73 static void Transform(const InType* in, const Quantize& params,
77 std::cout << "Quantize::Transform(" << std::string(typeid(InType).name())
88 template <typename InType, typename OutType, int kernel_size, int leftovers>
89 class Transform1DKernel<InType, OutType, Dequantize, kernel_size, leftovers> {
91 static void Transform(const InType* in, const Dequantize& params,
95 std::cout << "Dequantize::Transform(" << std::string(typeid(InType).name())
106 template <typename InType, typename OutType, int kernel_size, int leftovers>
107 class Transform1DKernel<InType, OutType, Requantize, kernel_size, leftovers>
[all...] |
| quantized_mul_kernels.h | 35 template <typename InType, typename OutType, int m, int n, int k>
36 class MulKernel<InType, OutType, QuantizedStaticPreprocessed, RowMajor, m, n,
41 static void Multiply(const InType* lhs, const InType*,
45 std::cout << "MulQSPR(" << typeid(InType).name() << ", "
61 std::cout << "MulQSPR(" << typeid(InType).name() << ", "
83 template <typename InType, typename OutType, int m, int n, int k>
84 class MulKernel<InType, OutType, QuantizedStaticPreprocessedAsInt32, RowMajor,
90 static void Multiply(const InType* lhs, const InType*,
[all...] |
| single_thread_gemm.h | 34 StreamUtil<typename P::InType, typename P::LeftStream>::Scratch(
39 StreamUtil<typename P::InType, typename P::RightStream>::Scratch(
48 typedef typename P::InType InType;
51 typedef Stream<typename P::InType, m, k, k_leftovers,
54 typedef Stream<typename P::InType, m_leftovers, k, k_leftovers,
58 typedef Stream<typename P::InType, n, k, k_leftovers,
61 typedef Stream<typename P::InType, n_leftovers, k, k_leftovers,
71 typedef MulKernel<typename P::InType, typename P::OutType,
74 typedef MulKernel<typename P::InType, typename P::OutType
[all...] |
| single_thread_transform.h | 33 Transform1DKernel<typename P::InType, typename P::OutType,
|
| multi_thread_transform.h | 32 typedef Transform1DUtil<typename Params::InType, typename Params::OutType,
|
| multi_thread_gemm.h | 39 StreamUtil<typename Params::InType, typename Params::LeftStream>::Offset(
44 StreamUtil<typename Params::InType, typename Params::RightStream>::Offset(
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| bfloat16.cc | 492 template <typename InType, typename OutType, typename Functor>
499 InType x = *reinterpret_cast<const InType*>(i0);
500 InType y = *reinterpret_cast<const InType*>(i1);
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_blas.cc | [all...] |
