Searched defs:num_batch (Results 1 – 3 of 3) sorted by relevance
138 int num_batch, OpKernelContext* ctx, in SetTrtEngineInputs()
781 const int num_batch = in ExecuteTrtEngine() local
1473 int64_t num_batch = input_type.getNumElements() / num_elems; in matchAndRewrite() local