Home
last modified time | relevance | path

Searched refs:RowPartitionTypeToString (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dragged_to_dense_util_test.cc89 TEST(RowPartitionTypeToString, BasicTest) { in TEST() argument
91 RowPartitionTypeToString(RowPartitionType::FIRST_DIM_SIZE)); in TEST()
93 RowPartitionTypeToString(RowPartitionType::VALUE_ROWIDS)); in TEST()
95 RowPartitionTypeToString(RowPartitionType::ROW_SPLITS)); in TEST()
H A Dragged_to_dense_util.h27 string RowPartitionTypeToString(RowPartitionType row_partition_type);
H A Dragged_to_dense_util_common.h33 inline std::string RowPartitionTypeToString( in RowPartitionTypeToString() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dragged_tensor_to_tensor_op.cc100 RowPartitionTypeToString( in GetMaxWidth()
344 RowPartitionTypeToString(partition_type)); in CalculateOutputIndex()
368 RowPartitionTypeToString(first_partition_type)); in GetFirstDimensionSize()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dragged_conversion_ops.cc40 RowPartitionTypeToString(row_partition_type)); in ValidateRowPartitionTypesAndShapes()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_tensor_to_tensor_tflite.cc80 RowPartitionTypeToString(first_partition_type).c_str()); in GetFirstDimensionSizeT()