Searched defs:TensorDataContainer (Results 1 – 1 of 1) sorted by relevance
109 struct TensorDataContainer { struct114 TensorDataContainer() in TensorDataContainer() function134 TensorDataContainer(std::initializer_list<TensorDataContainer> init_list) in AT_FORALL_SCALAR_TYPES_AND3() argument189 // 'torch::detail::TensorDataContainer' in AT_FORALL_SCALAR_TYPES_AND3() argument210 "Can only call `scalar()` on a TensorDataContainer that has `is_scalar() == true`"); in scalar() argument218 const std::initializer_list<TensorDataContainer>& init_list() const { in init_list() argument232 "Can only call `tensor()` on a TensorDataContainer that has `is_tensor() == true`"); in tensor() argument236 const std::vector<int64_t>& sizes() const { in sizes()240 const c10::ScalarType& scalar_type() const { in scalar_type()244 at::Tensor convert_to_tensor(at::TensorOptions options) const { in convert_to_tensor()[all …]