Searched refs:temp_prod (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 332 TfLiteTensor* temp_prod; in PrepareProd() local
334 GetTemporarySafe(context, node, /*index=*/2, &temp_prod)); in PrepareProd()
342 SetTensorToDynamic(temp_prod); in PrepareProd()
361 temp_prod->allocation_type = kTfLiteArenaRw; in PrepareProd()
362 return ResizeTempAccum(context, &op_context, temp_prod); in PrepareProd()
914 TfLiteTensor* temp_prod; in EvalQuantizedProd() local
916 GetTemporarySafe(context, node, /*index=*/2, &temp_prod)); in EvalQuantizedProd()
937 TF_LITE_ENSURE_OK(context, ResizeTempAccum(context, op_context, temp_prod)); in EvalQuantizedProd()
960 GetTensorData<int>(resolved_axis), GetTensorData<int32>(temp_prod), in EvalQuantizedProd()
973 GetTensorData<int32>(temp_prod), data->multiplier, data->shift)); in EvalQuantizedProd()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 479 int* resolved_axis, int32_t* temp_prod, in QuantizedReduceProd() argument
506 reducer_next, temp_prod)) { in QuantizedReduceProd()
512 MultiplyByQuantizedMultiplier(static_cast<int64_t>(temp_prod[i]), in QuantizedReduceProd()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | reduce.h | 625 int32_t* temp_prod, int32_t scaling_multiplier, int scaling_shift) { in QuantizedReduceProd() argument
641 num_resolved_axis, temp_prod, ReducerFirst<T>(input_zero_point), in QuantizedReduceProd()
649 MultiplyByQuantizedMultiplier(static_cast<int64_t>(temp_prod[i]), in QuantizedReduceProd()
|