Searched refs:input_depth (Results 1 – 2 of 2) sorted by relevance
75 const int input_depth = ArraySize(input_dims, 0); in quantizedLstmStep() local77 const int total_input_depth = prev_activ_depth + input_depth; in quantizedLstmStep()
2210 * The depth of the output tensor is input_depth * block_size * block_size.