Searched refs:buckets_t (Results 1 – 1 of 1) sorted by relevance
499 Tensor* buckets_t = nullptr; in Compute() local503 kBucketsName, {static_cast<int64>(boundaries.size())}, &buckets_t)); in Compute()504 auto* quantiles_flat = buckets_t->flat<float>().data(); in Compute()541 const Tensor* buckets_t; in Compute() local542 OP_REQUIRES_OK(context, context->input(kBucketsName, &buckets_t)); in Compute()565 const auto& buckets = buckets_t->vec<float>(); in Compute()