Lines Matching defs:InputType
42 template <typename OutputStage, typename InputType>
44 static constexpr int kRows = InputType::kRows;
45 static constexpr int kCols = InputType::kCols;
46 using InputBufferType = typename InputType::BufferType;
55 OutputType Eval(InputType input, int, int) const {
67 using InputType = RegisterBuffer<std::int32_t, Size>;
74 OutputType Eval(InputType input) const {
77 using RegisterType = typename InputType::RegisterType;
81 for (int i = 0; i < InputType::kRegisterCount; i++) {
94 typedef RegisterBlock<std::int32_t, Rows, Cols> InputType;
100 OutputType Eval(InputType input, int row, int col) const {
105 LoadForBroadcasting<InputType>(output_stage.result_mult_int, pos);
107 LoadForBroadcasting<InputType>(output_stage.result_offset, pos);
108 const auto dividend = BroadcastMul<InputType>(
109 BroadcastAdd<InputType>(input, result_offset), result_mult_int);
110 for (int i = 0; i < InputType::kRegisterCount; i++) {
124 typedef RegisterBuffer<std::int32_t, Size> InputType;
131 OutputType Eval(InputType input) const {
133 using RegisterType = typename InputType::RegisterType;
136 for (int i = 0; i < InputType::kRegisterCount; i++) {
152 typedef RegisterBuffer<std::int32_t, Size> InputType;
162 OutputType Eval(InputType input) const {
164 using RegisterType = typename InputType::RegisterType;
167 for (int i = 0; i < InputType::kRegisterCount; i++) {
186 typedef RegisterBuffer<std::int32_t, Size> InputType;
188 static_assert(InputType::kRegisterLanes == 1,
195 OutputType Eval(InputType input) const {
197 for (int i = 0; i < InputType::kRegisterCount; i++) {
209 typedef RegisterBuffer<std::int32_t, Size> InputType;
211 static_assert(InputType::kRegisterLanes == 1,
218 OutputType Eval(InputType input) const {
220 for (int i = 0; i < InputType::kRegisterCount; i++) {
231 typedef RegisterBlock<std::int32_t, Rows, Cols> InputType;
237 OutputType Eval(InputType input, int row, int col) const {
239 return BroadcastAdd<InputType>(
240 input, LoadForBroadcasting<InputType>(output_stage.bias_vector, pos));
249 typedef RegisterBuffer<std::int32_t, Size> InputType;
256 OutputType Eval(InputType input) const {
257 using RegisterType = typename InputType::RegisterType;
261 for (int i = 0; i < InputType::kRegisterCount; i++) {
273 typedef RegisterBuffer<std::int32_t, Size> InputType;
275 using RegisterType = typename InputType::RegisterType;
308 OutputType Eval(InputType input) const {
358 template <typename OutputPipelineType, int FirstStage, typename InputType,
364 typedef typename OutputStageEvalImpl<FirstStageType, InputType>::OutputType
370 template <typename OutputPipelineType, int FirstStage, typename InputType>
371 struct OutputPipelineOutputType<OutputPipelineType, FirstStage, InputType,
373 typedef InputType Type;
383 template <typename OutputPipelineType, int FirstStage, typename InputType,
389 typedef typename OutputStageEvalImpl<FirstStageType, InputType>::OutputType
392 InputType>::Type OutputType;
398 OutputType Eval(InputType input, int row, int col) const {
405 const OutputStageEvalImpl<FirstStageType, InputType> head_impl;
412 template <typename OutputPipelineType, int FirstStage, typename InputType>
413 struct OutputPipelineEvalImpl<OutputPipelineType, FirstStage, InputType, true> {
416 InputType Eval(InputType
450 template <typename OutputPipelineType, typename InputType>
460 void Execute(InputType input, DstType* dst, int src_global_row,
465 OutputPipelineType, 0, InputType>::Type::BufferType::ScalarType
479 const OutputPipelineEvalImpl<OutputPipelineType, 0, InputType>
