Searched refs:unary_op_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry.h | 115 const VariantUnaryOpFn& unary_op_fn);
268 UnaryVariantOpRegistry::VariantUnaryOpFn* unary_op_fn = in UnaryOpVariant() local
270 if (unary_op_fn == nullptr) { in UnaryOpVariant()
275 return (*unary_op_fn)(ctx, v, v_out); in UnaryOpVariant()
375 const LocalVariantUnaryOpFn& unary_op_fn) { in UnaryVariantUnaryOpRegistration() argument
379 [type_index_name, unary_op_fn](OpKernelContext* ctx, const Variant& v, in UnaryVariantUnaryOpRegistration()
390 return unary_op_fn(ctx, t, t_out); in UnaryVariantUnaryOpRegistration()
|
| D | variant_op_registry.cc | 170 const VariantUnaryOpFn& unary_op_fn) { in RegisterUnaryOpFn() argument
177 {op, GetPersistentStringPiece(device), type_index}, unary_op_fn)); in RegisterUnaryOpFn()
|