Commit 5ecab1ad authored by Sergey Shalnov's avatar Sergey Shalnov Committed by Robert Kimball

IntelGPU backend: fix typo in BatchNorm handling (#2294)

parent 25ab8a28
...@@ -1312,7 +1312,7 @@ runtime::Handle runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> ...@@ -1312,7 +1312,7 @@ runtime::Handle runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function>
arguments_check(op, 5, 1); arguments_check(op, 5, 1);
if (get_input_name(op, 2).size() != 4) if (get_input_shape(op, 2).size() != 4)
{ {
do_batch_norm_operation(topology, do_batch_norm_operation(topology,
get_output_name(op), get_output_name(op),
...@@ -1344,7 +1344,7 @@ runtime::Handle runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> ...@@ -1344,7 +1344,7 @@ runtime::Handle runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function>
static_pointer_cast<op::BatchNormTraining>(op); static_pointer_cast<op::BatchNormTraining>(op);
const double eps = bnorm->get_eps_value(); const double eps = bnorm->get_eps_value();
if (get_input_name(op, 2).size() != 4) if (get_input_shape(op, 2).size() != 4)
{ {
string mean_name; string mean_name;
string variance_name; string variance_name;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment