Searched refs:num_context_ragged (Results 1 – 5 of 5) sorted by relevance
392 int num_context_ragged, int num_feature_list_sparse, in TEST()428 DataTypeList(num_context_ragged, DT_FLOAT)) in TEST()430 DataTypeList(num_context_ragged, DT_INT32)) in TEST()
384 TF_RETURN_IF_ERROR(AddRaggedOutputShapes(attrs.num_context_ragged, false, in __anonb04296470802()
480 num_context_ragged = 0; in FinishInit()509 num_context_ragged = context_ragged_value_types.size(); in FinishInit()529 if ((num_context_ragged != context_ragged_value_types.size()) || in FinishInit()530 (num_context_ragged != context_ragged_split_types.size())) { in FinishInit()532 "num_context_ragged (", num_context_ragged, in FinishInit()
301 int64 num_context_ragged; member
510 if (context_ragged_keys->NumElements() != attrs_.num_context_ragged) { in CheckInputShapes()587 attrs_.num_context_ragged) in MakeContextConfig()605 config.ragged.reserve(attrs_.num_context_ragged); in MakeContextConfig()606 for (int d = 0; d < attrs_.num_context_ragged; ++d) { in MakeContextConfig()752 for (int d = 0; d < attrs_.num_context_ragged; ++d) { in WriteOutput()