Searched refs:stats_summary_t (Results 1 – 1 of 1) sorted by relevance
251 const Tensor* stats_summary_t; in Compute() local252 OP_REQUIRES_OK(context, context->input("stats_summary", &stats_summary_t)); in Compute()254 stats_summary_t->tensor<float, 4>(); in Compute()255 const int32 feature_dims = stats_summary_t->dim_size(1); in Compute()257 const int32 num_buckets = stats_summary_t->dim_size(2) - 1; in Compute()259 const int32 hessian_dim = stats_summary_t->dim_size(3) - logits_dim; in Compute()