Searched refs:pload4bf16 (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ | ||
H A D | sparse_matmul_op.h | 116 EIGEN_DEVICE_FUNC inline Packet pload4bf16( function in namespace:Eigen::internal 129 // Specialization for pload4bf16 and pload2bf16 for non-sse. 134 EIGEN_STRONG_INLINE Packet4f pload4bf16<Packet4f>(const float* from) { function in namespace:Eigen::internal 193 EIGEN_STRONG_INLINE Packet4f pload4bf16<Packet4f>(const float* from) { function in namespace:Eigen::internal 344 EIGEN_STRONG_INLINE Packet8f pload4bf16<Packet8f>(const float* from) { function in namespace:Eigen::internal 362 EIGEN_STRONG_INLINE Packet16f pload4bf16<Packet16f>(const float* from) { function in namespace:Eigen::internal |
H A D | sparse_matmul_op_test.cc | 412 internal::pstoreu(data2, internal::pload4bf16<Packet>(data3_bfloat16)); |
H A D | sparse_matmul_op.cc | 289 return Eigen::internal::pload4bf16<Packet>( |
Completed in 117 milliseconds