Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dspace_to_depth_test.cc42 ASSERT_OK(model.Invoke(*NewSpaceToDepthNodeShader())); in TEST()
57 ASSERT_OK(model.Invoke(*NewSpaceToDepthNodeShader())); in TEST()
75 ASSERT_OK(model.Invoke(*NewSpaceToDepthNodeShader())); in TEST()
94 ASSERT_OK(model.Invoke(*NewSpaceToDepthNodeShader())); in TEST()
Dspace_to_depth.h27 std::unique_ptr<NodeShader> NewSpaceToDepthNodeShader();
Dspace_to_depth.cc105 std::unique_ptr<NodeShader> NewSpaceToDepthNodeShader() { in NewSpaceToDepthNodeShader() function
Dregistry.cc102 insert_op(Type::SPACE_TO_DEPTH, NewSpaceToDepthNodeShader); in Registry()