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

  /external/tensorflow/tensorflow/core/kernels/
fused_batch_norm_op.cc 69 typename TTypes<U>::Vec batch_var(batch_var_output->vec<U>());
114 batch_var.device(d) = variance * rest_size_adjust;
225 Tensor* batch_mean, Tensor* batch_var, Tensor* saved_mean,
247 f(context->eigen_device<GPUDevice>(), batch_var->flat<U>());
305 auto batch_var_ptr = StreamExecutorUtil::AsDeviceMemory<U>(*batch_var);
556 Tensor* batch_var = nullptr; variable
558 context->allocate_output(2, scale.shape(), &batch_var));
568 y, batch_mean, batch_var, saved_mean, saved_maybe_inv_var,
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
virtual_scheduler_test.cc 248 auto batch_var = batch_norm.batch_variance; local
251 auto z2 = ops::Add(s.WithOpName("z2"), batch_var, batch_var);
252 auto z3 = ops::Add(s.WithOpName("z3"), batch_var, batch_var);
259 auto z4 = ops::NoOp(s.WithControlDependencies(batch_var).WithOpName("z4"));
811 auto batch_var = batch_norm.batch_variance; local
820 ops::Identity(s.WithOpName("batch_var1").WithDevice(kCPU1), batch_var);
    [all...]

Completed in 85 milliseconds