Searched refs:row_partition_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| 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()
304 const RowPartitionTensor row_partition_tensor = in CalculateOutputIndex() local
310 row_partition_tensor, parent_output_index, output_index_multiplier, in CalculateOutputIndex()
314 CalculateOutputIndexRowSplit(row_partition_tensor, parent_output_index, in CalculateOutputIndex()
|
| 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()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| 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()
|