载入中...
搜索中...
未找到
#include "tensor/Tensor.h"#include "tensor/CpuKernels.h"#include "tensor/Graph.h"#include "tensor/Quant.h"#include "common/Exception.h"#include <algorithm>#include <cmath>#include <cstring>命名空间 | |
| namespace | eve |
| namespace | eve::tensor |
宏定义 | |
| #define | EVE_TENSOR_UNARY(name, opType) |
函数 | |
| const char * | eve::tensor::dtypeName (DType dtype) |
| bool | eve::tensor::parseDType (const std::string &name, DType &out) |
宏定义说明
◆ EVE_TENSOR_UNARY
| #define EVE_TENSOR_UNARY | ( | name, | |
| opType | |||
| ) |
值:
Tensor *Tensor::name() const { \
ensureEager(#name); \
auto *out = new Tensor(dtype_, dims_, rank_); \
kernels::unaryOp((opType), data_.data(), size_, out->data_.data(), 0.f, 0.f); \
return out; \
}
在文件 Tensor.cpp 第 369 行定义.