Searched refs:num_ragged (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_array_ops.cc | 75 int num_ragged = ragged_values_types.size(); in __anoncd0b9d3b0102() local
76 if (num_ragged != ragged_splits_types.size()) { in __anoncd0b9d3b0102()
89 int ragged_splits_start = num_ragged; in __anoncd0b9d3b0102()
99 int dense_start = num_ragged * 2 + num_sparse * 3; in __anoncd0b9d3b0102()
|
| D | parsing_ops.cc | 65 Status AddRaggedOutputShapes(int num_ragged, bool ragged_rank_2, in AddRaggedOutputShapes() argument
71 for (int i = 0; i < num_ragged; ++i) { in AddRaggedOutputShapes()
75 for (int i = 0; i < num_ragged; ++i) { in AddRaggedOutputShapes()
80 for (int i = 0; i < num_ragged; ++i) { in AddRaggedOutputShapes()
221 TF_RETURN_IF_ERROR(AddRaggedOutputShapes(attrs.num_ragged, false, in __anonb04296470502()
|
| D | parsing_ops_test.cc | 305 auto set_outputs = [&op](int num_sparse, int num_dense, int num_ragged, in TEST()
322 .Attr("ragged_value_types", DataTypeList(num_ragged, DT_FLOAT)) in TEST()
323 .Attr("ragged_split_types", DataTypeList(num_ragged, DT_INT32)) in TEST()
|
| /external/tensorflow/tensorflow/python/util/ |
| D | example_parser_configuration.py | 147 num_ragged = len(ragged_value_types)
159 assert len(ragged_keys) == num_ragged
167 ragged_row_splits_start = ragged_values_start + num_ragged
202 if num_ragged != 0:
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ParseExampleV2.pbtxt | 75 A list of `num_ragged` types; the data types of data in each Feature
76 given in ragged_keys (where `num_ragged = sparse_keys.size()`).
84 A list of `num_ragged` types; the data types of row_splits in each Feature
85 given in ragged_keys (where `num_ragged = sparse_keys.size()`).
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.cc | 411 num_ragged = 0; in FinishInit()
415 num_ragged = ragged_value_types.size(); in FinishInit()
429 if (static_cast<size_t>(num_ragged) != ragged_value_types.size()) { in FinishInit()
433 if (static_cast<size_t>(num_ragged) != ragged_split_types.size()) { in FinishInit()
|
| D | example_proto_helper.h | 191 int64 num_ragged; member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_cross_op.cc | 382 const auto num_ragged = ragged_values_list.size(); in ValidateInput() local
386 for (int i = 0; i < num_ragged; ++i) { in ValidateInput()
417 for (int i = 0; i < num_ragged; ++i) { in ValidateInput()
|
| D | example_parsing_ops.cc | 161 if (ragged_keys_t.size() != attrs_.num_ragged) { in CheckInputShapes()
164 ragged_keys_t.size(), " vs. ", attrs_.num_ragged); in CheckInputShapes()
225 config.ragged.reserve(attrs_.num_ragged); in MakeConfig()
226 for (int d = 0; d < attrs_.num_ragged; ++d) { in MakeConfig()
278 for (int d = 0; d < attrs_.num_ragged; ++d) { in WriteOutput()
|