/external/tensorflow/tensorflow/core/kernels/ |
sparse_matmul_op_test.cc | 331 internal::pstoreu(data2, internal::pbroadcast_first<Packet>( 336 internal::pstoreu(data2, internal::pbroadcast_second<Packet>( 342 internal::pstoreu(data2, internal::pbroadcast_third<Packet>( 348 internal::pstoreu(data2, internal::pbroadcast_fourth<Packet>( 369 internal::pstoreu(data2, internal::pinterleave4x64<Packet>( 387 internal::pstoreu(data2, internal::pexpand_bf16_l<Packet>( 404 internal::pstoreu(data2, internal::pexpand_bf16_u<Packet>( 412 internal::pstoreu(data2, internal::pload4bf16<Packet>(data3_bfloat16)); 415 internal::pstoreu(data2, internal::pload2bf16<Packet>(data3_bfloat16));
|
eigen_spatial_convolutions.h | 731 pstoreu(block + 0 * packet_size, kernel.packet[0]); 732 pstoreu(block + 1 * packet_size, kernel.packet[1]); 733 pstoreu(block + 2 * packet_size, kernel.packet[2]); 734 pstoreu(block + 3 * packet_size, kernel.packet[3]); 748 pstoreu(block + 0 * packet_size, kernel.packet[0]); 749 pstoreu(block + 1 * packet_size, kernel.packet[1]); 750 pstoreu(block + 2 * packet_size, kernel.packet[2]); 751 pstoreu(block + 3 * packet_size, kernel.packet[3]); 762 pstoreu(block + 0 * packet_size, kernel.packet[0]); 763 pstoreu(block + 1 * packet_size, kernel.packet[1]) [all...] |
depthwise_conv_op.h | 152 Eigen::internal::pstoreu<T>(padded_filter + output_base + j, v); 253 Eigen::internal::pstoreu<T>(in_buf + base + dm, p);
|
depthwise_conv_grad_op.cc | 226 Eigen::internal::pstoreu<T>(buffer + buf_base + d, v); 314 Eigen::internal::pstoreu<T>(output + base_output_index + i, vaccum); 317 Eigen::internal::pstoreu<T>(out_buffer + i, vaccum); 332 Eigen::internal::pstoreu<T>(out_buf, vaccum); [all...] |
depthwise_conv_op.cc | 114 Eigen::internal::pstoreu<T>(output + base_output_index + i, vaccum); 131 Eigen::internal::pstoreu<T>(out_buf, vaccum);
|
sparse_matmul_op.h | 64 pstoreu(r, from); 77 pstoreu(r, from);
|
deep_conv2d.cc | 154 Eigen::internal::pstoreu<T>(filter_buf + d * kPacketSize, v); 245 ei::pstoreu<T>(filter_out + out_base + d * kPacketSize, v); 658 Eigen::internal::pstoreu<T>(tile, v); [all...] |
sparse_matmul_op.cc | [all...] |
/external/eigen/Eigen/src/Core/arch/ZVector/ |
Complex.h | 98 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<float> >(std::complex<float> * to, const Packet2cf& from) { EIGEN_DEBUG_UNALIGNED_STORE pstoreu((float*)to, from.v); } function in namespace:Eigen::internal 99 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<double> >(std::complex<double> * to, const Packet1cd& from) { EIGEN_DEBUG_UNALIGNED_STORE pstoreu((double*)to, from.v); } function in namespace:Eigen::internal
|
PacketMath.h | 676 template<> EIGEN_STRONG_INLINE void pstoreu<int>(int* to, const Packet4i& from) { pstore<int>(to, from); } function in namespace:Eigen::internal 677 template<> EIGEN_STRONG_INLINE void pstoreu<float>(float* to, const Packet4f& from) { pstore<float>(to, from); } function in namespace:Eigen::internal 678 template<> EIGEN_STRONG_INLINE void pstoreu<double>(double* to, const Packet2d& from) { pstore<double>(to, from); } function in namespace:Eigen::internal
|
/external/eigen/Eigen/src/Jacobi/ |
Jacobi.h | 374 pstoreu(px, padd(pm.pmul(pc,xi),pcj.pmul(ps,yi))); 375 pstoreu(px+PacketSize, padd(pm.pmul(pc,xi1),pcj.pmul(ps,yi1))); 385 pstoreu(x+peelingEnd, padd(pm.pmul(pc,xi),pcj.pmul(ps,yi)));
|
/external/eigen/Eigen/src/Core/arch/AVX/ |
Complex.h | 93 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<float> >(std::complex<float>* to, const Packet4cf& from) { EIGEN_DEBUG_UNALIGNED_STORE pstoreu(&numext::real_ref(*to), from.v); } function in namespace:Eigen::internal 311 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<double> >(std::complex<double> * to, const Packet2cd& from) { EIGEN_DEBUG_UNALIGNED_STORE pstoreu((double*)to, from.v); } function in namespace:Eigen::internal
|
PacketMath.h | 254 template<> EIGEN_STRONG_INLINE void pstoreu<float>(float* to, const Packet8f& from) { EIGEN_DEBUG_UNALIGNED_STORE _mm256_storeu_ps(to, from); } function in namespace:Eigen::internal 255 template<> EIGEN_STRONG_INLINE void pstoreu<double>(double* to, const Packet4d& from) { EIGEN_DEBUG_UNALIGNED_STORE _mm256_storeu_pd(to, from); } function in namespace:Eigen::internal 256 template<> EIGEN_STRONG_INLINE void pstoreu<int>(int* to, const Packet8i& from) { EIGEN_DEBUG_UNALIGNED_STORE _mm256_storeu_si256(reinterpret_cast<__m256i*>(to), from); } function in namespace:Eigen::internal
|
/external/eigen/Eigen/src/Core/arch/AltiVec/ |
Complex.h | 81 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<float> >(std::complex<float> * to, const Packet2cf& from) { pstoreu((float*)to, from.v); } function in namespace:Eigen::internal 310 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<double> >(std::complex<double> * to, const Packet1cd& from) { pstoreu((double*)to, from.v); } function in namespace:Eigen::internal
|
PacketMath.h | 466 template<> EIGEN_STRONG_INLINE void pstoreu<float>(float* to, const Packet4f& from) function in namespace:Eigen::internal 484 template<> EIGEN_STRONG_INLINE void pstoreu<int>(int* to, const Packet4i& from) function in namespace:Eigen::internal 504 template<> EIGEN_STRONG_INLINE void pstoreu<int>(int* to, const Packet4i& from) function in namespace:Eigen::internal 509 template<> EIGEN_STRONG_INLINE void pstoreu<float>(float* to, const Packet4f& from) function in namespace:Eigen::internal 943 template<> EIGEN_STRONG_INLINE void pstoreu<double>(double* to, const Packet2d& from) function in namespace:Eigen::internal [all...] |
/external/eigen/Eigen/src/Core/arch/NEON/ |
Complex.h | 127 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<float> >(std::complex<float> * to, const Packet2cf& from) { EIGEN_DEBUG_UNALIGNED_STORE pstoreu((float*)to, from.v); } function in namespace:Eigen::internal 382 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<double> >(std::complex<double> * to, const Packet1cd& from) { EIGEN_DEBUG_UNALIGNED_STORE pstoreu((double*)to, from.v); } function in namespace:Eigen::internal
|
PacketMath.h | 270 template<> EIGEN_STRONG_INLINE void pstoreu<float> (float* to, const Packet4f& from) { EIGEN_DEBUG_UNALIGNED_STORE vst1q_f32(to, from); } function in namespace:Eigen::internal 271 template<> EIGEN_STRONG_INLINE void pstoreu<int32_t>(int32_t* to, const Packet4i& from) { EIGEN_DEBUG_UNALIGNED_STORE vst1q_s32(to, from); } function in namespace:Eigen::internal 642 template<> EIGEN_STRONG_INLINE void pstoreu<double>(double* to, const Packet2d& from) { EIGEN_DEBUG_UNALIGNED_STORE vst1q_f64(to, from); } function in namespace:Eigen::internal
|
/external/eigen/Eigen/src/Core/arch/SSE/ |
Complex.h | 114 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<float> >(std::complex<float> * to, const Packet2cf& from) { EIGEN_DEBUG_UNALIGNED_STORE pstoreu(&numext::real_ref(*to), Packet4f(from.v)); } function in namespace:Eigen::internal 341 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<double> >(std::complex<double> * to, const Packet1cd& from) { EIGEN_DEBUG_UNALIGNED_STORE pstoreu((double*)to, Packet2d(from.v)); } function in namespace:Eigen::internal
|
PacketMath.h | 362 template<> EIGEN_STRONG_INLINE void pstoreu<double>(double* to, const Packet2d& from) { EIGEN_DEBUG_UNALIGNED_STORE _mm_storeu_pd(to, from); } function in namespace:Eigen::internal 363 template<> EIGEN_STRONG_INLINE void pstoreu<float>(float* to, const Packet4f& from) { EIGEN_DEBUG_UNALIGNED_STORE _mm_storeu_ps(to, from); } function in namespace:Eigen::internal 364 template<> EIGEN_STRONG_INLINE void pstoreu<int>(int* to, const Packet4i& from) { EIGEN_DEBUG_UNALIGNED_STORE _mm_storeu_si128(reinterpret_cast<__m128i*>(to), from); } function in namespace:Eigen::internal [all...] |
/external/eigen/Eigen/src/Core/ |
GenericPacketMath.h | 289 template<typename Scalar, typename Packet> EIGEN_DEVICE_FUNC inline void pstoreu(Scalar* to, const Packet& from) function in namespace:Eigen::internal 476 pstoreu(to, from);
|
/external/eigen/Eigen/src/Core/arch/CUDA/ |
PacketMath.h | 185 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void pstoreu<float>(float* to, const float4& from) { function in namespace:Eigen::internal 192 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void pstoreu<double>(double* to, const double2& from) { function in namespace:Eigen::internal
|
PacketMathHalf.h | 64 template<> __device__ EIGEN_STRONG_INLINE void pstoreu<Eigen::half>(Eigen::half* to, const half2& from) { function in namespace:Eigen::internal 403 template<> EIGEN_STRONG_INLINE void pstoreu<half>(Eigen::half* to, const Packet16h& from) { 771 template<> EIGEN_STRONG_INLINE void pstoreu<Eigen::half>(Eigen::half* to, const Packet8h& from) { [all...] |
/external/eigen/Eigen/src/Core/arch/AVX512/ |
PacketMath.h | 552 EIGEN_STRONG_INLINE void pstoreu<float>(float* to, const Packet16f& from) { function in namespace:Eigen::internal 556 EIGEN_STRONG_INLINE void pstoreu<double>(double* to, const Packet8d& from) { function in namespace:Eigen::internal 560 EIGEN_STRONG_INLINE void pstoreu<int>(int* to, const Packet16i& from) { function in namespace:Eigen::internal [all...] |
/external/eigen/Eigen/src/Core/products/ |
GeneralBlockPanelKernel.h | [all...] |
/external/eigen/test/ |
packetmath.cpp | 145 internal::pstoreu(data2+offset, internal::pload<Packet>(data1)); 146 VERIFY(areApprox(data1, data2+offset, PacketSize) && "internal::pstoreu");
|