Searched refs:tensor_buffer_allocator_ (Results 1 – 1 of 1) sorted by relevance
1246 : tensor_buffer_allocator_(new GpuManagedAllocator()), in OpConverterTest()1282 Tensor ret(tensor_buffer_allocator_.get(), DataTypeToEnum<T>::value, in AsTensor()1292 Tensor ret(tensor_buffer_allocator_.get(), DataTypeToEnum<T>::value, in AsTensor()1304 Tensor ret(tensor_buffer_allocator_.get(), tf_type, in AsTensor()1625 std::unique_ptr<Allocator> tensor_buffer_allocator_; member in tensorflow::tensorrt::convert::OpConverterTest