Lines Matching defs:mean
49 const Tensor& mean = context->input(1);
57 OP_REQUIRES(context, mean.dims() == 1,
58 errors::InvalidArgument("mean must be 1-dimensional",
59 mean.shape().DebugString()));
75 context->eigen_device<Device>(), input.tensor<T, 4>(), mean.vec<T>(),
99 const Tensor& mean = context->input(1);
107 OP_REQUIRES(context, mean.dims() == 1,
108 errors::InvalidArgument("mean must be 1-dimensional",
109 mean.shape().DebugString()));
125 {1}, 1, mean.shape(), &dm));
131 {3}, 3, mean.shape(), &db));
151 context->eigen_device<Device>(), input.tensor<T, 4>(), mean.vec<T>(),
181 typename TTypes<T>::ConstVec mean, typename TTypes<T>::ConstVec var, \
237 typename TTypes<T>::ConstVec mean, typename TTypes<T>::ConstVec var, \