Home
last modified time | relevance | path

Searched refs:num_context_dense (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()
163 .Input(NodeOutList(num_context_dense, node_in)) in TEST()
165 .Attr("Ncontext_dense", num_context_dense) in TEST()
173 std::vector<string>(num_context_dense, string_in)) in TEST()
181 DataTypeList(num_context_dense, DT_FLOAT)) in TEST()
183 MakeDenseShapes(num_context_dense, add_extra_shape, 0)) in TEST()
233 auto set_outputs = [&op](int num_context_sparse, int num_context_dense, in TEST()
246 .Input(NodeOutList(num_context_dense, string_in)) in TEST()
249 .Input(NodeOutList(num_context_dense, string_in)) in TEST()
254 DataTypeList(num_context_dense, DT_FLOAT)) in TEST()
[all …]
Dparsing_ops.cc193 for (int i = 0; i < attrs.num_context_dense; ++i) { in __anon952516500402()
288 for (int i = 0; i < attrs.num_context_dense; ++i) { in __anon952516500502()
/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops.cc296 OP_REQUIRES(ctx, context_dense_defaults.size() == attrs_.num_context_dense, in Compute()
300 attrs_.num_context_dense)); in Compute()
302 std::vector<bool> required(attrs_.num_context_dense); in Compute()
303 for (int d = 0; d < attrs_.num_context_dense; ++d) { in Compute()
327 for (int d = 0; d < attrs_.num_context_dense; ++d) { in Compute()
398 for (int d = 0; d < attrs_.num_context_dense; ++d) { in Compute()
458 std::vector<string> context_dense_keys_t(attrs_.num_context_dense); in Compute()
465 CHECK_EQ(context_dense_keys.size(), attrs_.num_context_dense); in Compute()
469 for (int di = 0; di < attrs_.num_context_dense; ++di) { in Compute()
537 OP_REQUIRES(ctx, context_dense_defaults.size() == attrs_.num_context_dense, in Compute()
[all …]
/external/tensorflow/tensorflow/python/ops/
Dparsing_ops.py1377 num_context_dense = len(context_dense_keys)
1382 if len(context_dense_shapes) != num_context_dense:
1385 (len(context_dense_shapes), num_context_dense))
1386 if len(context_dense_types) != num_context_dense:
1389 (len(context_dense_types), num_context_dense))
1408 if (num_context_dense + num_context_sparse + num_feature_list_dense +
1458 Ncontext_dense=num_context_dense,
1711 num_context_dense = len(context_dense_keys)
1716 if len(context_dense_shapes) != num_context_dense:
1719 % (len(context_dense_shapes), num_context_dense))
[all …]
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.h297 TF_RETURN_IF_ERROR(ctx->GetAttr("Ncontext_dense", &num_context_dense)); in Init()
317 int64 num_context_dense; variable
343 TF_RETURN_IF_ERROR(ctx->GetAttr("Ncontext_dense", &num_context_dense)); in Init()
362 int64 num_context_dense; variable
Dexample_proto_helper.cc455 if (num_context_dense != context_dense_keys.size() || in FinishInit()
456 num_context_dense != context_dense_types.size() || in FinishInit()
457 num_context_dense != context_dense_shapes.size()) { in FinishInit()
459 "num_context_dense (", num_context_dense, in FinishInit()
504 if (static_cast<size_t>(num_context_dense) != context_dense_types.size()) { in FinishInit()
508 if (static_cast<size_t>(num_context_dense) != context_dense_shapes.size()) { in FinishInit()