Searched defs:TensorBaseData (Results 1 – 2 of 2) sorted by relevance
122 struct TensorBaseData { struct123 TensorBaseData(uint64_t data_size, int dtype, const std::vector<int64_t> &shape) in TensorBaseData() argument126 const uint64_t data_size() const { return data_size_; } in data_size()127 const int dtype() const { return dtype_; } in dtype()128 const std::vector<int64_t> &shape() const { return shape_; } in shape()129 uint64_t data_size_;130 int dtype_;131 std::vector<int64_t> shape_;
616 class TensorBaseData: class