Searched refs:dims_with_batch (Results 1 – 1 of 1) sorted by relevance
1563 std::vector<int32> dims_with_batch(dims.size() + 1); in AddTestTensor() local1564 dims_with_batch[0] = batch_size; in AddTestTensor()1565 std::copy(dims.begin(), dims.end(), dims_with_batch.begin() + 1); in AddTestTensor()1566 AddTestTensorWithTFDims(name, dims_with_batch, trt_dtype); in AddTestTensor()