Searched refs:itensor (Results 1 – 2 of 2) sorted by relevance
40 Tensor<int, 2> itensor(20,30); in test_vectorized_cast() local41 itensor = itensor.random() / 1000; in test_vectorized_cast()47 ftensor = itensor.cast<float>(); in test_vectorized_cast()48 dtensor = itensor.cast<double>(); in test_vectorized_cast()52 VERIFY_IS_EQUAL(itensor(i,j), static_cast<int>(ftensor(i,j))); in test_vectorized_cast()
471 FakeITensor itensor(dims); in TEST() local472 TRT_TensorOrWeights tw(&itensor); in TEST()473 TRT_TensorOrWeights tw1(&itensor, /*batch_size=*/1); in TEST()488 EXPECT_EQ(&itensor, ptr->tensor()); in TEST()