Home
last modified time | relevance | path

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

/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc60 int GetFirstDimensionSizeT(TfLiteContext* context, in GetFirstDimensionSizeT() function
90 return GetFirstDimensionSizeT<int32_t>(context, first_partition_input, in GetFirstDimensionSize()
93 return GetFirstDimensionSizeT<int64_t>(context, first_partition_input, in GetFirstDimensionSize()