Searched refs:maxvals_tensor (Results 1 – 2 of 2) sorted by relevance
333 const Tensor& maxvals_tensor = ctx->input(4); in Compute() local369 OP_REQUIRES(ctx, maxvals_tensor.dims() <= 1, in Compute()372 maxvals_tensor.shape().DebugString())); in Compute()376 minvals_tensor.dims() == 0 && maxvals_tensor.dims() == 0) { in Compute()414 TensorShapeUtils::IsScalar(maxvals_tensor.shape()) || in Compute()415 maxvals_tensor.dim_size(0) == 1 || in Compute()416 maxvals_tensor.dim_size(0) == num_batches, in Compute()419 maxvals_tensor.shape().DebugString())); in Compute()430 maxvals_tensor.flat<T>(), rng, samples_tensor->flat<T>()); in Compute()
128 maxvals_tensor = ops.convert_to_tensor(maxvals, dtype=dtype, name="maxvals")135 maxvals_tensor,