Home
last modified time | relevance | path

Searched refs:num_context_dense (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dparsing_ops_test.cc154 auto set_outputs = [&op](int num_context_sparse, int num_context_dense, in TEST()
166 .Input(NodeOutList(num_context_dense, node_in)) in TEST()
168 .Attr("Ncontext_dense", num_context_dense) in TEST()
176 std::vector<string>(num_context_dense, string_in)) in TEST()
184 DataTypeList(num_context_dense, DT_FLOAT)) in TEST()
186 MakeDenseShapes(num_context_dense, add_extra_shape, 0)) in TEST()
236 auto set_outputs = [&op](int num_context_sparse, int num_context_dense, in TEST()
249 .Input(NodeOutList(num_context_dense, string_in)) in TEST()
252 .Input(NodeOutList(num_context_dense, string_in)) in TEST()
257 DataTypeList(num_context_dense, DT_FLOAT)) in TEST()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops.cc502 if (context_dense_keys->NumElements() != attrs_.num_context_dense) { in CheckInputShapes()
540 if (context_dense_defaults.size() != attrs_.num_context_dense) { in CheckInputShapes()
544 context_dense_defaults.size(), " vs. ", attrs_.num_context_dense); in CheckInputShapes()
546 for (int d = 0; d < attrs_.num_context_dense; ++d) { in CheckInputShapes()
577 attrs_.num_context_dense) in MakeContextConfig()
591 config.dense.reserve(attrs_.num_context_dense); in MakeContextConfig()
592 for (int d = 0; d < attrs_.num_context_dense; ++d) { in MakeContextConfig()
719 for (int d = 0; d < attrs_.num_context_dense; ++d) { in WriteOutput()
809 std::vector<string> context_dense_keys_t(attrs_.num_context_dense); in Compute()
824 CHECK_EQ(context_dense_keys.size(), attrs_.num_context_dense); in Compute()
[all …]
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.cc488 if (num_context_dense != context_dense_keys.size()) { in FinishInit()
490 "num_context_dense (", num_context_dense, in FinishInit()
508 num_context_dense = context_dense_types.size(); in FinishInit()
521 if (num_context_dense != context_dense_types.size() || in FinishInit()
522 num_context_dense != context_dense_shapes.size()) { in FinishInit()
524 "num_context_dense (", num_context_dense, in FinishInit()
599 if (static_cast<size_t>(num_context_dense) != context_dense_types.size()) { in FinishInit()
603 if (static_cast<size_t>(num_context_dense) != context_dense_shapes.size()) { in FinishInit()
Dexample_proto_helper.h264 TF_RETURN_IF_ERROR(ctx->GetAttr("Ncontext_dense", &num_context_dense));
300 int64 num_context_dense; member
332 TF_RETURN_IF_ERROR(ctx->GetAttr("Ncontext_dense", &num_context_dense)); in Init()
351 int64 num_context_dense; member