Searched refs:AccumT (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | redux_functor.h | 34 template <typename T, typename AccumT, typename BinaryFunctor>
68 Tensor buffer(DataTypeToEnum<AccumT>::v(), {num_blocks, inner_dim}); in operator()
69 buffer.template flat<AccumT>().setZero(); in operator()
70 AccumT* buffer_data = buffer.template flat<AccumT>().data(); in operator()
73 Eigen::Tensor<AccumT, 1, Eigen::RowMajor, Eigen::Index>, in operator()
91 auto cast = in.template cast<AccumT>(); in operator()
|
| D | bias_op.cc | 290 using AccumT = typename AccumulatorType<T>::type; in Compute() typedef
292 T, AccumT, Eigen::internal::scalar_sum_op<AccumT>> in Compute()
|