Searched refs:TensorDataContainerType (Results 1 – 1 of 1) sorted by relevance
23 enum class TensorDataContainerType { Scalar, InitList, Tensor }; enum121 type_(TensorDataContainerType::InitList) {}126 type_(TensorDataContainerType::Scalar), \137 type_(TensorDataContainerType::InitList), in AT_FORALL_SCALAR_TYPES_AND3()169 type_(TensorDataContainerType::Tensor) { \204 return type_ == TensorDataContainerType::Scalar;215 return type_ == TensorDataContainerType::InitList; in is_init_list()226 return type_ == TensorDataContainerType::Tensor; in is_tensor()348 TensorDataContainerType type_;