Home
last modified time | relevance | path

Searched defs:first_dimension (Results 1 – 3 of 3) 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()
373 INDEX_TYPE first_dimension; in Compute() local
/external/ComputeLibrary/arm_compute/core/
DWindow.inl276 inline void Window::use_tensor_dimensions(const TensorShape &shape, size_t first_dimension)