Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.cc161 UnaryVariantOpRegistry::VariantUnaryOpFn* UnaryVariantOpRegistry::GetUnaryOpFn( in GetUnaryOpFn()
170 const VariantUnaryOpFn& unary_op_fn) { in RegisterUnaryOpFn()
171 VariantUnaryOpFn* existing = GetUnaryOpFn(op, device, type_index); in RegisterUnaryOpFn()
176 unary_op_fns.insert(std::pair<FuncTuple<VariantUnaryOp>, VariantUnaryOpFn>( in RegisterUnaryOpFn()
Dvariant_op_registry.h63 VariantUnaryOpFn; typedef
115 const VariantUnaryOpFn& unary_op_fn);
119 VariantUnaryOpFn* GetUnaryOpFn(VariantUnaryOp op, StringPiece device,
204 gtl::FlatMap<FuncTuple<VariantUnaryOp>, VariantUnaryOpFn, TupleHash>
268 UnaryVariantOpRegistry::VariantUnaryOpFn* unary_op_fn = in UnaryOpVariant()
Dvariant_op_registry_test.cc231 UnaryVariantOpRegistry::VariantUnaryOpFn f; in TEST()