| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_lstm.cc | 1248 use_aux_input ? GetTemporary(context, node, kAuxInputQuantized) in Eval()
1289 fw_scratch_buffer, GetTemporary(context, node, kInputScalingFactors), in Eval()
1290 GetTemporary(context, node, kAuxInputScalingFactors), in Eval()
1291 GetTemporary(context, node, kOutputStateScalingFactors), in Eval()
1296 GetTemporary(context, node, kInputZeroPoints), in Eval()
1297 GetTemporary(context, node, kAuxInputZeroPoints), in Eval()
1298 GetTemporary(context, node, kOutputStateZeroPoints), fw_row_sums, in Eval()
1330 bw_scratch_buffer, GetTemporary(context, node, kInputScalingFactors), in Eval()
1331 GetTemporary(context, node, kAuxInputScalingFactors), in Eval()
1332 GetTemporary(context, node, kOutputStateScalingFactors), in Eval()
[all …]
|
| D | lstm.cc | 2015 /*output_offset=*/0, GetTemporary(context, node, kScratchBuffer), in Eval()
2016 GetTemporary(context, node, kInputScalingFactors), in Eval()
2018 GetTemporary(context, node, kOutputStateScalingFactors), in Eval()
2019 GetTemporary(context, node, kProductScalingFactors), in Eval()
2020 GetTemporary(context, node, kRecoveredCellWeights), in Eval()
2021 GetTemporary(context, node, kInputQuantized), in Eval()
2023 GetTemporary(context, node, kOutputStateQuantized), in Eval()
2024 GetTemporary(context, node, kCellStateQuantized), output_state, in Eval()
2025 cell_state, GetTemporary(context, node, kAccumScratch), output, in Eval()
2026 GetTemporary(context, node, kInputZeroPoints), in Eval()
[all …]
|
| D | unidirectional_sequence_lstm.cc | 1382 GetTemporary(context, node, kInputScalingFactors), in Eval()
1384 GetTemporary(context, node, kOutputStateScalingFactors), in Eval()
1385 GetTemporary(context, node, kProductScalingFactors), in Eval()
1386 GetTemporary(context, node, kRecoveredCellWeights), in Eval()
1387 GetTemporary(context, node, kInputQuantized), in Eval()
1389 GetTemporary(context, node, kOutputStateQuantized), in Eval()
1390 GetTemporary(context, node, kCellStateQuantized), output_state, in Eval()
1391 cell_state, GetTemporary(context, node, kAccumScratch), output, in Eval()
1392 GetTemporary(context, node, kInputZeroPoints), in Eval()
1394 GetTemporary(context, node, kOutputStateZeroPoints), row_sums, in Eval()
|
| D | batch_matmul.cc | 611 TfLiteTensor* transposed_rhs = GetTemporary(context, node, 1); in GetTempRhs()
626 TfLiteTensor* transposed_lhs = GetTemporary(context, node, 0); in GetTempLhs()
675 TransposeRowsColumns(context, rhs, GetTemporary(context, node, 1)); in Eval()
680 TransposeRowsColumns(context, lhs, GetTemporary(context, node, 0)); in Eval()
|
| D | kernel_util.h | 105 TfLiteTensor* GetTemporary(TfLiteContext* context, const TfLiteNode* node,
|
| D | transpose_conv.cc | 585 ? GetTemporary(context, node, data->col2im_index) in Eval()
589 ? GetTemporary(context, node, data->transposed_weights_index) in Eval()
|
| D | kernel_util.cc | 145 TfLiteTensor* GetTemporary(TfLiteContext* context, const TfLiteNode* node, in GetTemporary() function
|
| D | bidirectional_sequence_rnn.cc | 819 use_aux_input ? GetTemporary(context, node, kAuxInputQuantized) in Eval()
|
| /external/angle/src/compiler/translator/tree_ops/ |
| D | RewriteDfdy.cpp | 151 const TType *vec2Type = StaticType::GetTemporary<EbtFloat, 2>(); in visitAggregateWithRotation()
|
| /external/angle/src/compiler/translator/ |
| D | StaticType.h | 204 constexpr const TType *GetTemporary() in GetTemporary() function
|
| D | TranslatorVulkan.cpp | 310 const TType *vec2Type = StaticType::GetTemporary<EbtFloat, 2>(); in AddBresenhamEmulationVS()
659 const TType *vec2Type = StaticType::GetTemporary<EbtFloat, 2>(); in AddBresenhamEmulationFS()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 592 TfLiteTensor* scratch_buffer = GetTemporary(context, node, /*index=*/0); in lstm_eval()
|