Searched refs:num_batch (Results 1 – 2 of 2) sorted by relevance
401 const int num_batch = ctx->input(0).shape().dim_size(0); in ExecuteTrtEngine() local417 if (num_batch != input_shape.dim_size(0)) { in ExecuteTrtEngine()418 LOG(ERROR) << "Input data has inconsistent batch size: " << num_batch in ExecuteTrtEngine()454 trt_shape[0] = num_batch; in ExecuteTrtEngine()509 auto ret = engine_context->execution_context->enqueue(num_batch, &buffers[0], in ExecuteTrtEngine()
871 num_batch = 10883 train_samples=(batch * num_batch),