Searched refs:stddevs_tensor (Results 1 – 2 of 2) sorted by relevance
331 const Tensor& stddevs_tensor = ctx->input(2); in Compute() local361 OP_REQUIRES(ctx, stddevs_tensor.dims() <= 1, in Compute()364 stddevs_tensor.shape().DebugString())); in Compute()375 (stddevs_tensor.dims() == 0 || stddevs_tensor.dim_size(0) == 1) && in Compute()398 TensorShapeUtils::IsScalar(stddevs_tensor.shape()) || in Compute()399 stddevs_tensor.dim_size(0) == 1 || in Compute()400 stddevs_tensor.dim_size(0) == num_batches, in Compute()403 stddevs_tensor.shape().DebugString())); in Compute()429 stddevs_tensor.flat<T>(), minvals_tensor.flat<T>(), in Compute()
126 stddevs_tensor = ops.convert_to_tensor(stddevs, dtype=dtype, name="stddevs")133 stddevs_tensor,