Searched refs:AccumT (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | redux_functor.h | 38 template <typename InputT, typename AccumT, typename OutputT,
80 Eigen::Tensor<AccumT, 1, Eigen::RowMajor, Eigen::Index> buffer( in operator()
83 AccumT* buffer_data = buffer.data(); in operator()
86 Eigen::Tensor<AccumT, 1, Eigen::RowMajor, Eigen::Index>, in operator()
107 auto cast = in.template cast<AccumT>(); in operator()
147 Tensor buffer(DataTypeToEnum<AccumT>::v(), {num_blocks, inner_dim}); in operator()
148 buffer.template flat<AccumT>().setZero(); in operator()
149 AccumT* buffer_data = buffer.template flat<AccumT>().data(); in operator()
152 Eigen::Tensor<AccumT, 1, Eigen::RowMajor, Eigen::Index>, in operator()
170 auto cast = in.template cast<AccumT>(); in operator()
[all …]
|
| D | bias_op.cc | 258 using AccumT = typename AccumulatorType<T>::type; in Compute() typedef
261 T, AccumT, T, Eigen::internal::scalar_sum_op<AccumT>, in Compute()
270 T, AccumT, T, Eigen::internal::scalar_sum_op<AccumT>> in Compute()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_matmul.h | 198 template <typename T, typename AccumT>
251 AccumT total = 0; in BatchMatMul()
253 AccumT lhs_val = lhs_ptr2[accum_depth * i + k]; in BatchMatMul()
254 AccumT rhs_val = rhs_ptr2[accum_depth * j + k]; in BatchMatMul()
|