Searched refs:tensor_buffer_allocator_ (Results 1 – 1 of 1) sorted by relevance
1357 : tensor_buffer_allocator_(new GpuManagedAllocator()), in OpConverterTest()1391 Tensor ret(tensor_buffer_allocator_.get(), DataTypeToEnum<T>::value, in AsTensor()1401 Tensor ret(tensor_buffer_allocator_.get(), DataTypeToEnum<T>::value, in AsTensor()1413 Tensor ret(tensor_buffer_allocator_.get(), tf_type, in AsTensor()1713 std::unique_ptr<Allocator> tensor_buffer_allocator_; member in tensorflow::tensorrt::convert::OpConverterTest