aten_compiler.h File Reference

#include <chrono>
#include <string>
#include <vector>
#include <ATen/ATen.h>
#include <ATen/DLConvertor.h>
#include "tc/core/execution_engine.h"
#include "tc/core/utils/cuda_info.h"
#include "tc/lang/parser.h"

Include dependency graph for aten_compiler.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tc::ATenCompilationUnit

Namespaces
	tc

Functions
std::pair< std::vector < DLTensor * >, std::vector < DLManagedTensor * > >	tc::toDlpackTensors (const std::vector< at::Tensor > &tensors)
std::pair< std::vector< const DLTensor * >, std::vector < DLManagedTensor * > >	tc::toConstDlpackTensors (const std::vector< at::Tensor > &tensors)
void	tc::deleteDlmTensors (std::vector< DLManagedTensor * > &tensors)