Searched refs:num_feature_list_ragged (Results 1 – 5 of 5) sorted by relevance
394 int num_feature_list_ragged, in TEST()432 DataTypeList(num_feature_list_ragged, DT_FLOAT)) in TEST()434 DataTypeList(num_feature_list_ragged, DT_INT32)) in TEST()
395 attrs.num_feature_list_ragged, true, num_examples, c, &output_idx)); in __anonb04296470802()
481 num_feature_list_ragged = 0; in FinishInit()510 num_feature_list_ragged = feature_list_ragged_value_types.size(); in FinishInit()551 if ((num_feature_list_ragged != feature_list_ragged_value_types.size()) || in FinishInit()552 (num_feature_list_ragged != feature_list_ragged_split_types.size())) { in FinishInit()554 "num_feature_list_ragged (", num_feature_list_ragged, in FinishInit()
304 int64 num_feature_list_ragged; member
534 attrs_.num_feature_list_ragged) { in CheckInputShapes()646 attrs_.num_feature_list_ragged) in MakeFeatureListConfig()676 config.ragged.reserve(attrs_.num_feature_list_ragged); in MakeFeatureListConfig()677 for (int d = 0; d < attrs_.num_feature_list_ragged; ++d) { in MakeFeatureListConfig()756 for (int d = 0; d < attrs_.num_feature_list_ragged; ++d) { in WriteOutput()