Searched defs:FlattenWithTensorOp (Results 1 – 1 of 1) sorted by relevance
398 struct FlattenWithTensorOp : public torch::CustomClassHolder { struct399 explicit FlattenWithTensorOp(at::Tensor t) : t_(t) {} in FlattenWithTensorOp() function401 at::Tensor get() { in get()405 std::tuple<std::tuple<std::string, at::Tensor>> __obj_flatten__() { in __obj_flatten__()410 at::Tensor t_;