Searched refs:num_context_sparse (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | parsing_ops_test.cc | 154 auto set_outputs = [&op](int num_context_sparse, int num_context_dense, in TEST()
167 .Attr("Ncontext_sparse", num_context_sparse) in TEST()
174 std::vector<string>(num_context_sparse, string_in)) in TEST()
182 DataTypeList(num_context_sparse, DT_FLOAT)) in TEST()
236 auto set_outputs = [&op](int num_context_sparse, int num_context_dense, in TEST()
248 .Input(NodeOutList(num_context_sparse, string_in)) in TEST()
255 DataTypeList(num_context_sparse, DT_FLOAT)) in TEST()
391 auto set_outputs = [&op](int num_context_sparse, int num_context_dense, in TEST()
412 .Attr("Ncontext_sparse", num_context_sparse) in TEST()
416 DataTypeList(num_context_sparse, DT_FLOAT)) in TEST()
|
| D | parsing_ops.cc | 296 AddSparseOutputShapes(attrs.num_context_sparse, input, 1, c, &output_idx); in __anonf8f6e66e0702()
381 AddSparseOutputShapes(attrs.num_context_sparse, input, 1, c, &output_idx); in __anonf8f6e66e0802()
442 AddSparseOutputShapes(attrs.num_context_sparse, input, 1, c, &output_idx); in __anonf8f6e66e0902()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.h | 283 TF_RETURN_IF_ERROR(ctx->GetAttr("Ncontext_sparse", &num_context_sparse));
299 int64 num_context_sparse; member
335 TF_RETURN_IF_ERROR(ctx->GetAttr("Ncontext_sparse", &num_context_sparse)); in Init()
350 int64 num_context_sparse; member
|
| D | example_proto_helper.cc | 482 if (num_context_sparse != context_sparse_keys.size()) { in FinishInit()
484 "num_context_sparse (", num_context_sparse, in FinishInit()
515 if (num_context_sparse != context_sparse_types.size()) { in FinishInit()
517 "num_context_sparse (", num_context_sparse, in FinishInit()
595 if (static_cast<size_t>(num_context_sparse) != context_sparse_types.size()) { in FinishInit()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 506 if (context_sparse_keys->NumElements() != attrs_.num_context_sparse) { in CheckInputShapes()
582 attrs_.num_context_sparse) in MakeContextConfig()
600 config.sparse.reserve(attrs_.num_context_sparse); in MakeContextConfig()
601 for (int d = 0; d < attrs_.num_context_sparse; ++d) { in MakeContextConfig()
726 for (int d = 0; d < attrs_.num_context_sparse; ++d) { in WriteOutput()
810 std::vector<string> context_sparse_keys_t(attrs_.num_context_sparse); in Compute()
825 CHECK_EQ(context_sparse_keys.size(), attrs_.num_context_sparse); in Compute()
837 for (int di = 0; di < attrs_.num_context_sparse; ++di) { in Compute()
1016 for (int d = 0; d < attrs_.num_context_sparse; ++d) { in Compute()
|