Searched defs:num_batch (Results 1 – 4 of 4) sorted by relevance
57 const uint32_t num_batch = src.shape()[1]; in qlstm_layer_normalization() local
100 int num_batch, in SetTrtEngineInputs()
868 const int num_batch = in ExecuteTrtEngine() local
1325 int64_t num_batch = input_type.getNumElements() / num_elems; in matchAndRewrite() local