Skip to content

Commit

Permalink
Fix code style
Browse files Browse the repository at this point in the history
  • Loading branch information
mvafin committed Aug 11, 2023
1 parent 1656c55 commit a54ef36
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions src/frontends/pytorch/src/op/rand.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -254,7 +254,8 @@ OutputVector translate_randn_like(const NodeContext& context) {
};

OutputVector translate_randint(const NodeContext& context) {
// aten::randint.low(int low, int high, SymInt[] size, *, ScalarType? dtype=4, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
// aten::randint.low(int low, int high, SymInt[] size, *, ScalarType? dtype=4, Layout? layout=None, Device?
// device=None, bool? pin_memory=None) -> Tensor
num_inputs_check(context, 7, 7);
auto low = context.get_input(0);
auto high = context.get_input(1);
Expand All @@ -265,7 +266,8 @@ OutputVector translate_randint(const NodeContext& context) {
if (!context.input_is_none(3)) {
if (std::dynamic_pointer_cast<v0::Constant>(context.get_input_from_visible_context(3).get_node_shared_ptr())) {
dtype = convert_dtype(context.const_input<int64_t>(3));
} else if (const auto& fw_node = cast_fw_node(context.get_input(static_cast<int>(3)).get_node_shared_ptr(), "prim::dtype")) {
} else if (const auto& fw_node =
cast_fw_node(context.get_input(static_cast<int>(3)).get_node_shared_ptr(), "prim::dtype")) {
convert_like_out = fw_node->input_value(0);
dtype_applied = false;
} else {
Expand Down

0 comments on commit a54ef36

Please sign in to comment.