HomeSort by relevance Sort by last modified time
    Searched refs:saved_var (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_fused_batch_norm_op.cc 364 const Tensor& saved_var = MklGetInput(context, 4); variable
390 OP_REQUIRES(context, saved_var.dims() == 1,
392 saved_var.shape().DebugString()));
639 const Tensor& saved_var = MklGetInput(context, 4); local
643 static_cast<const void*>(saved_var.flat<T>().data()));
    [all...]
fused_batch_norm_op.cc 71 typename TTypes<U>::Vec saved_var(saved_var_output->vec<U>());
115 saved_var.device(d) = variance;
    [all...]

Completed in 118 milliseconds