Searched refs:parent_output_index (Results 1 – 3 of 3) 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()
263 const vector<INDEX_TYPE>& parent_output_index, in CalculateOutputIndexValueRowID() argument
274 DCHECK_LT(current_value_rowid, parent_output_index.size()); in CalculateOutputIndexValueRowID()
275 INDEX_TYPE current_output_index = parent_output_index[current_value_rowid]; in CalculateOutputIndexValueRowID()
291 DCHECK_LT(next_value_rowid, parent_output_index.size()); in CalculateOutputIndexValueRowID()
292 current_output_index = parent_output_index[next_value_rowid]; in CalculateOutputIndexValueRowID()
300 const vector<INDEX_TYPE>& parent_output_index, in CalculateOutputIndex() argument
310 row_partition_tensor, parent_output_index, output_index_multiplier, in CalculateOutputIndex()
314 CalculateOutputIndexRowSplit(row_partition_tensor, parent_output_index, in CalculateOutputIndex()
|
| /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/lite/python/ |
| D | op_hint.py | 1243 parent_output_index = _get_correct_mapping(
1248 parent_output_index]
|