get layout from get_output_layout in primitive_inst (#21294)
This commit is contained in:
parent
e52f922d35
commit
1a235ffe79
@ -76,7 +76,7 @@ struct condition_impl : typed_primitive_impl<condition> {
|
||||
if (instance.dependencies()[dep_idx].first->id() == input_external_node->first) {
|
||||
output_events.push_back(events[dep_idx]);
|
||||
output_mem_ptr = instance.input_memory_ptr(dep_idx);
|
||||
output_layout = output_mem_ptr->get_layout();
|
||||
output_layout = instance.dependencies()[dep_idx].first->get_output_layout();
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user