Home
last modified time | relevance | path

Searched refs:GetUnaryOpFn (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.cc161 UnaryVariantOpRegistry::VariantUnaryOpFn* UnaryVariantOpRegistry::GetUnaryOpFn( in GetUnaryOpFn() function in tensorflow::UnaryVariantOpRegistry
171 VariantUnaryOpFn* existing = GetUnaryOpFn(op, device, type_index); in RegisterUnaryOpFn()
Dvariant_op_registry_test.cc179 EXPECT_EQ(UnaryVariantOpRegistry::Global()->GetUnaryOpFn( in TEST()
205 EXPECT_EQ(UnaryVariantOpRegistry::Global()->GetUnaryOpFn( in TEST()
Dvariant_op_registry.h119 VariantUnaryOpFn* GetUnaryOpFn(VariantUnaryOp op, StringPiece device,
269 UnaryVariantOpRegistry::Global()->GetUnaryOpFn(op, device, v.TypeId()); in UnaryOpVariant()