Searched refs:num_bits_tensor (Results 1 – 1 of 1) sorted by relevance
261 const Tensor num_bits_tensor = ctx->input(3); in Compute() local262 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(num_bits_tensor.shape()), in Compute()265 num_bits_tensor.dims())); in Compute()267 const int num_bits_val = num_bits_tensor.scalar<int32>()(); in Compute()