Searched refs:num_bits_tensor (Results 1 – 1 of 1) sorted by relevance
231 Tensor num_bits_tensor; in Compute() local232 num_bits_tensor = ctx->input(3); in Compute()233 int num_bits_val = num_bits_tensor.scalar<int32>()(); in Compute()