Searched refs:num_buckets_t (Results 1 – 1 of 1) sorted by relevance
346 const Tensor* num_buckets_t; in Compute() local347 OP_REQUIRES_OK(context, context->input(kNumBucketsName, &num_buckets_t)); in Compute()348 const int64 num_buckets = num_buckets_t->scalar<int64>()(); in Compute()