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

/external/tensorflow/tensorflow/core/kernels/
H A Dgather_nd_op_test.cc44 class Node* GatherNd(Graph* g, class Node* in0, class Node* in1) { function in namespace:tensorflow::test::graph
46 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "GatherNd")
61 TF_ASSERT_OK(NodeDefBuilder("myop", "GatherNd")
86 static Graph* GatherNd(int dim) { function in namespace:tensorflow::__anon26483
104 test::graph::GatherNd(g, test::graph::Constant(g, params),
115 test::Benchmark(#DEVICE, GatherNd<INDEX>(dim)).Run(iters); \

Completed in 153 milliseconds