Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dragged_to_dense_util_common.h28 ROW_SPLITS, enumerator
42 case RowPartitionType::ROW_SPLITS: in RowPartitionTypeToString()
60 {"ROW_SPLITS", RowPartitionType::ROW_SPLITS}, in GetRowPartitionTypesHelper()
Dragged_to_dense_util_test.cc86 RowPartitionType::ROW_SPLITS)); in TEST()
95 RowPartitionTypeToString(RowPartitionType::ROW_SPLITS)); in TEST()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_RaggedTensorToTensor.pbtxt8 * "ROW_SPLITS": the row_splits tensor from the ragged tensor.
76 * "ROW_SPLITS": the row_splits tensor from the ragged tensor.
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_tensor_op.cc94 case RowPartitionType::ROW_SPLITS: in GetMaxWidth()
313 case RowPartitionType::ROW_SPLITS: in CalculateOutputIndex()
336 case RowPartitionType::ROW_SPLITS: in GetFirstDimensionSize()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc72 case tensorflow::RowPartitionType::ROW_SPLITS: { in GetFirstDimensionSizeT()
220 case tensorflow::RowPartitionType::ROW_SPLITS: in GetMaxWidth()
438 case tensorflow::RowPartitionType::ROW_SPLITS: in CalculateOutputIndex()
/external/tensorflow/tensorflow/core/ops/
Dragged_conversion_ops.cc36 case RowPartitionType::ROW_SPLITS: in ValidateRowPartitionTypesAndShapes()