Home
last modified time | relevance | path

Searched refs:NewReLUNodeShader (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Drelu_test.cc53 ASSERT_OK(model.Invoke(*NewReLUNodeShader())); in TEST_F()
66 ASSERT_OK(model.Invoke(*NewReLUNodeShader())); in TEST_F()
79 ASSERT_OK(model.Invoke(*NewReLUNodeShader())); in TEST_F()
92 ASSERT_OK(model.Invoke(*NewReLUNodeShader())); in TEST_F()
Drelu.h28 std::unique_ptr<NodeShader> NewReLUNodeShader();
Drelu.cc74 std::unique_ptr<NodeShader> NewReLUNodeShader() { in NewReLUNodeShader() function
Dregistry.cc96 insert_op(Type::RELU, NewReLUNodeShader); in Registry()