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

  /external/tensorflow/tensorflow/core/kernels/
mkl_fused_batch_norm_op.cc 154 Tensor* batch_variance = nullptr; variable
157 AllocateOutputSetMklShape(context, 2, &batch_variance, scale.shape(),
164 mkl_context.MklSetMeanVariance(*batch_mean, *batch_variance);
196 float* p_var = static_cast<float*>(batch_variance->flat<T>().data());
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
virtual_scheduler_test.cc 248 auto batch_var = batch_norm.batch_variance;
811 auto batch_var = batch_norm.batch_variance;
    [all...]

Completed in 62 milliseconds