Home
last modified time | relevance | path

Searched refs:num_context_sparse (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dparsing_ops_test.cc151 auto set_outputs = [&op](int num_context_sparse, int num_context_dense, in TEST()
164 .Attr("Ncontext_sparse", num_context_sparse) in TEST()
171 std::vector<string>(num_context_sparse, string_in)) in TEST()
179 DataTypeList(num_context_sparse, DT_FLOAT)) in TEST()
233 auto set_outputs = [&op](int num_context_sparse, int num_context_dense, in TEST()
245 .Input(NodeOutList(num_context_sparse, string_in)) in TEST()
252 DataTypeList(num_context_sparse, DT_FLOAT)) in TEST()
Dparsing_ops.cc182 for (int i = 0; i < attrs.num_context_sparse; ++i) { in __anon952516500402()
185 for (int i = 0; i < attrs.num_context_sparse; ++i) { in __anon952516500402()
188 for (int i = 0; i < attrs.num_context_sparse; ++i) { in __anon952516500402()
277 for (int i = 0; i < attrs.num_context_sparse; ++i) { in __anon952516500502()
280 for (int i = 0; i < attrs.num_context_sparse; ++i) { in __anon952516500502()
283 for (int i = 0; i < attrs.num_context_sparse; ++i) { in __anon952516500502()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.h300 TF_RETURN_IF_ERROR(ctx->GetAttr("Ncontext_sparse", &num_context_sparse)); in Init()
316 int64 num_context_sparse; variable
346 TF_RETURN_IF_ERROR(ctx->GetAttr("Ncontext_sparse", &num_context_sparse)); in Init()
361 int64 num_context_sparse; variable
Dexample_proto_helper.cc447 if (num_context_sparse != context_sparse_keys.size() || in FinishInit()
448 num_context_sparse != context_sparse_types.size()) { in FinishInit()
450 "num_context_sparse (", num_context_sparse, in FinishInit()
500 if (static_cast<size_t>(num_context_sparse) != context_sparse_types.size()) { in FinishInit()
/external/tensorflow/tensorflow/python/ops/
Dparsing_ops.py1379 num_context_sparse = len(context_sparse_keys)
1399 if len(context_sparse_types) != num_context_sparse:
1402 (len(context_sparse_types), num_context_sparse))
1408 if (num_context_dense + num_context_sparse + num_feature_list_dense +
1457 Ncontext_sparse=num_context_sparse,
1713 num_context_sparse = len(context_sparse_keys)
1733 if len(context_sparse_types) != num_context_sparse:
1736 % (len(context_sparse_types), num_context_sparse))
1742 if (num_context_dense + num_context_sparse
/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops.cc334 for (int d = 0; d < attrs_.num_context_sparse; ++d) { in Compute()
407 for (int d = 0; d < attrs_.num_context_sparse; ++d) { in Compute()
459 std::vector<string> context_sparse_keys_t(attrs_.num_context_sparse); in Compute()
466 CHECK_EQ(context_sparse_keys.size(), attrs_.num_context_sparse); in Compute()
478 for (int di = 0; di < attrs_.num_context_sparse; ++di) { in Compute()
660 for (int d = 0; d < attrs_.num_context_sparse; ++d) { in Compute()