Home
last modified time | relevance | path

Searched refs:GetRowPartitionTypeByDimension (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dragged_tensor_to_tensor_op.cc69 RowPartitionType GetRowPartitionTypeByDimension(int dimension) { in GetRowPartitionTypeByDimension() function in tensorflow::__anonf14c6a980111::RaggedTensorToTensorBaseOp
90 switch (GetRowPartitionTypeByDimension(dimension - 1)) { in GetMaxWidth()
101 GetRowPartitionTypeByDimension(dimension - 1))); in GetMaxWidth()
325 auto partition_type = GetRowPartitionTypeByDimension(dimension); in CalculateOutputIndex()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_tensor_to_tensor_tflite.cc49 tensorflow::RowPartitionType GetRowPartitionTypeByDimension( in GetRowPartitionTypeByDimension() function
217 switch (conversion_attributes.GetRowPartitionTypeByDimension(dimension - 1)) { in GetMaxWidth()
431 conversion_attributes.GetRowPartitionTypeByDimension(dimension); in CalculateOutputIndex()