Home
last modified time | relevance | path

Searched defs:prev_activ_depth (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dlstm_cell.h78 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell() local
283 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DQuantizedLSTM.cpp77 const int prev_activ_depth = ArraySize(prev_activ_dims, 0); in quantizedLstmStep() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dlegacy_optimized_ops.h3029 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell() local
3175 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell() local
H A Doptimized_ops.h2501 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell() local
2626 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell() local