Searched refs:bucket_boundaries_t (Results 1 – 2 of 2) sorted by relevance
299 const Tensor& bucket_boundaries_t = bucket_boundaries_list[stream_idx]; in Compute() local300 const auto& bucket_boundaries = bucket_boundaries_t.vec<float>(); in Compute()411 Tensor* bucket_boundaries_t = nullptr; in Compute() local415 &bucket_boundaries_t)); in Compute()416 auto* quantiles_flat = bucket_boundaries_t->flat<float>().data(); in Compute()
145 const Tensor* bucket_boundaries_t; in Compute() local147 context->input("bucket_boundaries", &bucket_boundaries_t)); in Compute()148 const auto& bucket_boundaries = bucket_boundaries_t->vec<float>(); in Compute()432 const Tensor* bucket_boundaries_t; in Compute() local434 context->input("bucket_boundaries", &bucket_boundaries_t)); in Compute()435 const auto& bucket_boundaries = bucket_boundaries_t->vec<float>(); in Compute()