Home
last modified time | relevance | path

Searched defs:first_dimension (Results 1 – 2 of 2) sorted by relevance

/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc251 TfLiteNode* node, int first_dimension, int ragged_rank, in CalculateOutputSize()
295 void CalculateFirstParentOutputIndex(int first_dimension, in CalculateFirstParentOutputIndex()
625 const int first_dimension = in Eval() local
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_tensor_op.cc192 void CalculateFirstParentOutputIndex(INDEX_TYPE first_dimension, in CalculateFirstParentOutputIndex()
347 INDEX_TYPE first_dimension; in Compute() local