Searched refs:parent_output_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_tensor_op.cc | 212 const vector<INDEX_TYPE>& parent_output_index, in CalculateOutputIndexRowSplit() argument
222 INDEX_TYPE parent_output_index_current = parent_output_index[i]; in CalculateOutputIndexRowSplit()
265 const vector<INDEX_TYPE>& parent_output_index, in CalculateOutputIndexValueRowID() argument
277 if (current_value_rowid >= parent_output_index.size()) { in CalculateOutputIndexValueRowID()
280 " which is not less than ", parent_output_index.size()); in CalculateOutputIndexValueRowID()
283 INDEX_TYPE current_output_index = parent_output_index[current_value_rowid]; in CalculateOutputIndexValueRowID()
300 if (next_value_rowid >= parent_output_index.size()) { in CalculateOutputIndexValueRowID()
303 " which is not less than ", parent_output_index.size()); in CalculateOutputIndexValueRowID()
306 current_output_index = parent_output_index[next_value_rowid]; in CalculateOutputIndexValueRowID()
319 const vector<INDEX_TYPE>& parent_output_index, in CalculateOutputIndex() argument
[all …]
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_tensor_to_tensor_tflite.cc | 332 const std::vector<int>& parent_output_index, in CalculateOutputIndexValueRowID() argument
357 int current_output_index = parent_output_index[current_value_rowid]; in CalculateOutputIndexValueRowID()
374 current_output_index = parent_output_index[next_value_rowid]; in CalculateOutputIndexValueRowID()
382 const std::vector<int>& parent_output_index, in CalculateOutputIndexRowSplit() argument
405 int parent_output_index_current = parent_output_index[i]; in CalculateOutputIndexRowSplit()
426 const std::vector<int>& parent_output_index, int output_index_multiplier, in CalculateOutputIndex() argument
434 CalculateOutputIndexValueRowID(*row_partition_tensor, parent_output_index, in CalculateOutputIndex()
439 CalculateOutputIndexRowSplit(*row_partition_tensor, parent_output_index, in CalculateOutputIndex()
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-048.md | 42 INDEX_TYPE parent_output_index_current = parent_output_index[i];
48 trigger a heap OOB access when `parent_output_index` is shorter than
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | op_hint.py | 1224 parent_output_index = _get_correct_mapping(
1229 parent_output_index]
|