Searched refs:row_partition_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | ragged_tensor_to_tensor_op.cc | 88 const RowPartitionTensor row_partition_tensor = in GetMaxWidth() local
92 *result = GetMaxWidthValueRowID(row_partition_tensor); in GetMaxWidth()
95 *result = GetMaxWidthRowSplit(row_partition_tensor); in GetMaxWidth()
323 const RowPartitionTensor row_partition_tensor = in CalculateOutputIndex() local
329 row_partition_tensor, parent_output_index, output_index_multiplier, in CalculateOutputIndex()
332 if (row_partition_tensor.size() - 1 > parent_output_index.size()) { in CalculateOutputIndex()
335 row_partition_tensor.size() - 1, " > ", in CalculateOutputIndex()
339 row_partition_tensor, parent_output_index, output_index_multiplier, in CalculateOutputIndex()
|
| H A D | ragged_tensor_to_tensor_op_test.cc | 87 for (const auto& row_partition_tensor : row_partition_tensors) { in BuildRaggedTensorToTensorGraph() local
88 AddInputFromArray<INDEX_TYPE>(row_partition_tensor.shape, in BuildRaggedTensorToTensorGraph()
89 row_partition_tensor.values); in BuildRaggedTensorToTensorGraph()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| H A D | ragged_tensor_to_tensor_tflite.cc | 428 const TfLiteTensor* row_partition_tensor = in CalculateOutputIndex() local
434 CalculateOutputIndexValueRowID(*row_partition_tensor, parent_output_index, in CalculateOutputIndex()
439 CalculateOutputIndexRowSplit(*row_partition_tensor, parent_output_index, in CalculateOutputIndex()
|