Home
last modified time | relevance | path

Searched defs:row_partition_tensor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_tensor_op.cc88 const RowPartitionTensor row_partition_tensor = in GetMaxWidth() local
304 const RowPartitionTensor row_partition_tensor = in CalculateOutputIndex() local
Dragged_tensor_to_tensor_op_test.cc87 for (const auto& row_partition_tensor : row_partition_tensors) { in BuildRaggedTensorToTensorGraph() local
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc428 const TfLiteTensor* row_partition_tensor = in CalculateOutputIndex() local