HomeSort by relevance Sort by last modified time
    Searched defs:scale_backprop (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
batch_norm_op.cc 147 xla::ComputationDataHandle scale_backprop; variable
155 scale_backprop = b->GetTupleElement(output, 1);
165 // scale_backprop = y_backprop * ((x - pop_mean) * rsqrt(pop_var +
185 scale_backprop = b->Mul(scratch1, scratch2);
189 ctx->SetOutput(1, scale_backprop);
  /external/tensorflow/tensorflow/core/kernels/
fused_batch_norm_op.cc 150 typename TTypes<U>::Vec scale_backprop(scale_backprop_output->vec<U>());
158 // scale_backprop = sum(y_backprop *
193 scale_backprop.device(d) =
367 Tensor* scale_backprop, Tensor* offset_backprop,
457 StreamExecutorUtil::AsDeviceMemory<U>(*scale_backprop);
629 Tensor* scale_backprop = nullptr; variable
631 &scale_backprop));
651 f(context->eigen_device<Device>(), scale_backprop->flat<U>());
659 saved_maybe_inv_var_or_pop_var, epsilon_, x_backprop, scale_backprop,
678 x_backprop, scale_backprop, offset_backprop, scratch1.vec<U>()
    [all...]
mkl_fused_batch_norm_op.cc 450 Tensor* scale_backprop = nullptr; variable
452 AllocateOutputSetMklShape(context, 1, &scale_backprop, tf_shape_scale_shift,
465 static_cast<const float*>(scale_backprop->flat<T>().data()));
    [all...]

Completed in 157 milliseconds