Searched defs:buckets_list (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
H A Dquantile_ops.cc80 const OpInputList& buckets_list) {
81 const auto& buckets = buckets_list[feature].flat<float>();
100 const OpInputList& buckets_list,
124 GetBuckets(feature_index, buckets_list);
690 OpOutputList buckets_list; variable
691 OP_REQUIRES_OK(context, context->output_list(kBucketsName, &buckets_list));
702 &buckets_list, stamp_token](int64 start, int64 end) {
728 OP_REQUIRES_OK(context, buckets_list.allocate(
79 GetBuckets(const int32 feature, const OpInputList& buckets_list) argument
98 QuantizeFeatures( const string& output_name, const OpInputList& values_list, const OpInputList& buckets_list, const OpInputList* const indices_list , OpKernelContext* const context) argument

Completed in 85 milliseconds