Home
last modified time | relevance | path

Searched defs:GatherNd (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgather_nd_op_test.cc44 class Node* GatherNd(Graph* g, class Node* in0, class Node* in1) { in GatherNd() function
86 static Graph* GatherNd(int dim) { in GatherNd() function
/external/tensorflow/tensorflow/lite/kernels/
Dgather_nd.cc97 TfLiteStatus GatherNd(const TfLiteTensor* params, const TfLiteTensor* indices, in GatherNd() function
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc549 class GatherNd class
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h2973 inline void GatherNd(const RuntimeShape& params_shape, in GatherNd() function
/external/tensorflow/tensorflow/go/op/
Dwrappers.go2264 func GatherNd(scope *Scope, params tf.Output, indices tf.Output) (output tf.Output) { func