Searched refs:num_feature_list_sparse (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | parsing_ops_test.cc | 155 int num_feature_list_sparse, in TEST()
169 .Attr("Nfeature_list_sparse", num_feature_list_sparse) in TEST()
178 std::vector<string>(num_feature_list_sparse, string_in)) in TEST()
188 DataTypeList(num_feature_list_sparse, DT_FLOAT)) in TEST()
237 int num_feature_list_sparse, in TEST()
250 .Input(NodeOutList(num_feature_list_sparse, string_in)) in TEST()
261 DataTypeList(num_feature_list_sparse, DT_FLOAT)) in TEST()
392 int num_context_ragged, int num_feature_list_sparse, in TEST()
413 .Attr("Nfeature_list_sparse", num_feature_list_sparse) in TEST()
422 DataTypeList(num_feature_list_sparse, DT_FLOAT)) in TEST()
|
| H A D | parsing_ops.cc | 301 AddSparseOutputShapes(attrs.num_feature_list_sparse, input, 2, c, in __anonf9b685fd0702()
389 AddSparseOutputShapes(attrs.num_feature_list_sparse, input, 2, c, in __anonf9b685fd0802()
447 AddSparseOutputShapes(attrs.num_feature_list_sparse, input, 2, c, in __anonf9b685fd0902()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | example_proto_helper.h | 291 ctx->GetAttr("Nfeature_list_sparse", &num_feature_list_sparse));
303 int64_t num_feature_list_sparse; member
343 ctx->GetAttr("Nfeature_list_sparse", &num_feature_list_sparse)); in Init()
353 int64_t num_feature_list_sparse; member
|
| H A D | example_proto_helper.cc | 494 if (num_feature_list_sparse != feature_list_sparse_keys.size()) { in FinishInit()
496 "num_feature_list_sparse (", num_feature_list_sparse, in FinishInit()
537 if (num_feature_list_sparse != feature_list_sparse_types.size()) { in FinishInit()
539 "num_feature_list_sparse (", num_feature_list_sparse, in FinishInit()
607 if (static_cast<size_t>(num_feature_list_sparse) != in FinishInit()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | example_parsing_ops.cc | 528 attrs_.num_feature_list_sparse) { in CheckInputShapes()
641 attrs_.num_feature_list_sparse) in MakeFeatureListConfig()
671 config.sparse.reserve(attrs_.num_feature_list_sparse); in MakeFeatureListConfig()
672 for (int d = 0; d < attrs_.num_feature_list_sparse; ++d) { in MakeFeatureListConfig()
731 for (int d = 0; d < attrs_.num_feature_list_sparse; ++d) { in WriteOutput()
814 attrs_.num_feature_list_sparse); in Compute()
827 CHECK_EQ(feature_list_sparse_keys.size(), attrs_.num_feature_list_sparse); in Compute()
856 for (int di = 0; di < attrs_.num_feature_list_sparse; ++di) { in Compute()
1122 for (int d = 0; d < attrs_.num_feature_list_sparse; ++d) { in Compute()
|