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>
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,
get_output_name(op),
......@@ -1344,7 +1344,7 @@ runtime::Handle runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function>
static_pointer_cast<op::BatchNormTraining>(op);
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 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