Home
last modified time | relevance | path

Searched refs:m_tensor (Results 1 – 4 of 4) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorContractionMapper.h42 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE CoeffLoader(const Tensor& tensor) : m_tensor(tensor) { } in CoeffLoader()
54 …typename Tensor::Scalar coeff(typename Tensor::Index index) const { return m_tensor.coeff(index); } in coeff()
59 return m_tensor.template packet<LoadMode>(index); in packet()
65 m_tensor.bind(cgh); in bind()
70 const Tensor m_tensor;
122 m_tensor(tensor),
133 m_tensor.offsetBuffer(offset);
147 return m_tensor.coeff(computeIndex(row, col));
254 m_tensor.bind(cgh);
259 return m_tensor;
[all …]
/external/armnn/src/armnnOnnxParser/
DOnnxParser.hpp144 std::unique_ptr<const onnx::TensorProto> m_tensor; member
147 OnnxTensor() : m_info(nullptr), m_tensor(nullptr), m_dtype(onnx::TensorProto::FLOAT) { } in OnnxTensor()
148 bool isConstant() { return m_tensor != nullptr; } in isConstant()
DOnnxParser.cpp467 onnx::TensorProto onnxTensor = *m_TensorsInfo[name].m_tensor; in CreateConstTensor()
624 m_TensorsInfo[tensor.name()].m_tensor = std::make_unique<const onnx::TensorProto>(tensor); in LoadGraph()
1265 m_TensorsInfo[node.output(0)].m_tensor = std::make_unique<const onnx::TensorProto>(onnxTensor); in ParseConstant()
1571 m_TensorsInfo[node.output(0)].m_tensor = in ParseReshape()
1572 std::make_unique<onnx::TensorProto>(*m_TensorsInfo[node.input(0)].m_tensor); in ParseReshape()
1580 … uint64_t dims = static_cast<uint64_t>(m_TensorsInfo[node.input(1)].m_tensor->int64_data_size()); in ParseReshape()
1585 … int val = CHECKED_INT32(m_TensorsInfo[node.input(1)].m_tensor->int64_data(static_cast<int>(i))); in ParseReshape()
1631 …m_TensorsInfo[outputName].m_tensor = std::make_unique<onnx::TensorProto>(*m_TensorsInfo[input0].m_… in PrependForBroadcast()
/external/eigen/unsupported/Eigen/CXX11/src/TensorSymmetry/
DSymmetry.h305 : m_tensor(tensor), m_symmetry(symmetry), m_indices(indices) { }
313 Tensor_& m_tensor;
326 …emplate apply<internal::tensor_symmetry_assign_value<Tensor_>, int>(m_indices, 0, m_tensor, value);