Searched defs:TensorCastValue (Results 1 – 1 of 1) sorted by relevance
545 struct TORCH_API TensorCastValue : public SugaredValue { struct546 TensorCastValue(at::ScalarType type, NamedValue self) in TensorCastValue() argument549 std::string kind() const override { in kind()553 std::shared_ptr<SugaredValue> call( in call()571 at::ScalarType dtype_;572 NamedValue self_;