Searched defs:GatherNd (Results 1 – 5 of 5) sorted by relevance
44 class Node* GatherNd(Graph* g, class Node* in0, class Node* in1) { in GatherNd() function86 static Graph* GatherNd(int dim) { in GatherNd() function
97 TfLiteStatus GatherNd(const TfLiteTensor* params, const TfLiteTensor* indices, in GatherNd() function
549 class GatherNd class
2973 inline void GatherNd(const RuntimeShape& params_shape, in GatherNd() function
2264 func GatherNd(scope *Scope, params tf.Output, indices tf.Output) (output tf.Output) { func