Searched refs:batch_bucketized_features (Results 1 – 1 of 1) sorted by relevance
68 std::vector<tensorflow::TTypes<int32>::ConstVec> batch_bucketized_features; in Compute() local69 batch_bucketized_features.reserve(bucketized_features_list.size()); in Compute()71 batch_bucketized_features.emplace_back(tensor.vec<int32>()); in Compute()73 const int batch_size = batch_bucketized_features[0].size(); in Compute()114 auto do_work = [&resource, &batch_bucketized_features, &cached_tree_ids, in Compute()158 batch_bucketized_features); in Compute()207 std::vector<tensorflow::TTypes<int32>::ConstVec> batch_bucketized_features; in Compute() local208 batch_bucketized_features.reserve(bucketized_features_list.size()); in Compute()210 batch_bucketized_features.emplace_back(tensor.vec<int32>()); in Compute()212 const int batch_size = batch_bucketized_features[0].size(); in Compute()[all …]