Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.h67 VariantUnaryOpFn; typedef
133 const VariantUnaryOpFn& unary_op_fn) { in RegisterUnaryOpFn()
134 VariantUnaryOpFn* existing = GetUnaryOpFn(op, device, type_index); in RegisterUnaryOpFn()
139 unary_op_fns.insert(std::pair<FuncTuple<VariantUnaryOp>, VariantUnaryOpFn>( in RegisterUnaryOpFn()
145 VariantUnaryOpFn* GetUnaryOpFn(VariantUnaryOp op, StringPiece device, in GetUnaryOpFn()
251 gtl::FlatMap<FuncTuple<VariantUnaryOp>, VariantUnaryOpFn, TupleHash>
315 UnaryVariantOpRegistry::VariantUnaryOpFn* unary_op_fn = in UnaryOpVariant()
Dvariant_op_registry_test.cc248 UnaryVariantOpRegistry::VariantUnaryOpFn f; in TEST()