| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| blas_gemm.h | 40 struct gemm_compute_type { struct in namespace:tensorflow::functor
45 struct gemm_compute_type<Eigen::half> { struct in namespace:tensorflow::functor
55 bool transb, typename gemm_compute_type<T>::type alpha,
58 typename gemm_compute_type<T>::type beta,
73 bool transb, typename gemm_compute_type<T>::type alpha,
76 typename gemm_compute_type<T>::type beta,
81 if (alpha == typename gemm_compute_type<T>::type(1.f) &&
82 beta == typename gemm_compute_type<T>::type(0.f)) {
84 } else if (alpha == typename gemm_compute_type<T>::type(1.f) &&
85 beta == typename gemm_compute_type<T>::type(1.f))
[all...] |
| gru_ops.h | 91 ctx, d, false, false, typename gemm_compute_type<T>::type(1.f),
92 const_x_h_prev, w_ru, typename gemm_compute_type<T>::type(0.f),
112 ctx, d, false, false, typename gemm_compute_type<T>::type(1.f),
113 const_x_h_prevr, w_c, typename gemm_compute_type<T>::type(0.f), c);
155 ctx, d, false, true, typename gemm_compute_type<T>::type(1.f),
156 const_d_c_bar, w_c, typename gemm_compute_type<T>::type(0.f),
171 ctx, d, false, true, typename gemm_compute_type<T>::type(1.f),
172 const_d_r_bar_u_bar, w_ru, typename gemm_compute_type<T>::type(0.f),
|
| lstm_ops_gpu.cu.cc | 252 ctx, d, false, false, typename gemm_compute_type<T>::type(1.f), const_xh,
253 w, typename gemm_compute_type<T>::type(0.f), icfo);
|
| lstm_ops.cc | 64 ctx, d, false, false, typename gemm_compute_type<T>::type(1.f), const_xh,
65 w, typename gemm_compute_type<T>::type(0.f), icfo);
[all...] |
