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.hpp186 std::unique_ptr<const onnx::TensorProto> m_tensor; member
189 OnnxTensor() : m_info(nullptr), m_tensor(nullptr), m_dtype(onnx::TensorProto::FLOAT) { } in OnnxTensor()
190 bool isConstant() { return m_tensor != nullptr; } in isConstant()
DOnnxParser.cpp605 onnx::TensorProto onnxTensor = *m_TensorsInfo[name].m_tensor; in CreateConstTensor()
651 onnx::TensorProto onnxTensor = *m_TensorsInfo[name].m_tensor; in CreateInt64ConstTensor()
891 m_TensorsInfo[tensor.name()].m_tensor = std::make_unique<const onnx::TensorProto>(tensor); in LoadGraph()
1701 m_TensorsInfo[node.output(0)].m_tensor = std::make_unique<const onnx::TensorProto>(onnxTensor); in ParseConstant()
2225 …unsigned int dims = static_cast<unsigned int>(m_TensorsInfo[node.input(1)].m_tensor->int64_data_si… in ParseReshape()
2230 … int val = CHECKED_INT32(m_TensorsInfo[node.input(1)].m_tensor->int64_data(static_cast<int>(i))); in ParseReshape()
2265 m_TensorsInfo[node.output(0)].m_tensor = in ParseReshape()
2266 std::make_unique<onnx::TensorProto>(*m_TensorsInfo[node.input(0)].m_tensor); in ParseReshape()
2299 auto int64Axes = m_TensorsInfo[node.input(1)].m_tensor->int64_data().data(); in ParseUnsqueeze()
2300 … uint numDim = armnn::numeric_cast<uint>(m_TensorsInfo[node.input(1)].m_tensor->int64_data_size()); in ParseUnsqueeze()
[all …]
/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);