Searched refs:means_tensor (Results 1 – 2 of 2) sorted by relevance
330 const Tensor& means_tensor = ctx->input(1); in Compute() local357 OP_REQUIRES(ctx, means_tensor.dims() <= 1, in Compute()360 means_tensor.shape().DebugString())); in Compute()374 if ((means_tensor.dims() == 0 || means_tensor.dim_size(0) == 1) && in Compute()390 TensorShapeUtils::IsScalar(means_tensor.shape()) || in Compute()391 means_tensor.dim_size(0) == 1 || in Compute()392 means_tensor.dim_size(0) == num_batches, in Compute()395 means_tensor.shape().DebugString())); in Compute()428 samples_per_batch, num_elements, means_tensor.flat<T>(), in Compute()
125 means_tensor = ops.convert_to_tensor(means, dtype=dtype, name="means")132 means_tensor,