Searched refs:RegisterUnaryOpFn (Results 1 – 3 of 3) sorted by relevance
235 registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_CPU, in TEST()237 EXPECT_DEATH(registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, in TEST()241 registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, in TEST()243 EXPECT_DEATH(registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, in TEST()
113 void RegisterUnaryOpFn(VariantUnaryOp op, const string& device,377 UnaryVariantOpRegistry::Global()->RegisterUnaryOpFn( in UnaryVariantUnaryOpRegistration()
168 void UnaryVariantOpRegistry::RegisterUnaryOpFn( in RegisterUnaryOpFn() function in tensorflow::UnaryVariantOpRegistry