summaryrefslogtreecommitdiffstats
path: root/eval/src/vespa/eval/tensor/dense/dense_tensor_function_compiler.h
blob: ef940bf38f94acbbad903f1e6dde91d33cb922b7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#pragma once

#include <vespa/eval/eval/tensor_function.h>

namespace vespalib {
namespace tensor {

/**
 * Class that recognizes calculations over dense tensors (in tensor function intermediate representation)
 * and compiles this into an explicit tensor function.
 */
struct DenseTensorFunctionCompiler
{
    static eval::TensorFunction::UP compile(eval::tensor_function::Node_UP expr);
};

} // namespace tensor
} // namespace vespalib