Searched refs:tensor_buffer_allocator_ (Results 1 – 1 of 1) sorted by relevance
1128 : tensor_buffer_allocator_(new GpuManagedAllocator()), in OpConverterTest()1167 Tensor ret(tensor_buffer_allocator_.get(), DataTypeToEnum<T>::value, in AsTensor()1177 Tensor ret(tensor_buffer_allocator_.get(), DataTypeToEnum<T>::value, in AsTensor()1202 Tensor ret(tensor_buffer_allocator_.get(), tf_type, in AsTensor()1587 std::unique_ptr<Allocator> tensor_buffer_allocator_; member in tensorflow::tensorrt::convert::OpConverterTest