Searched refs:unary_op_fn (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry.h | 133 const VariantUnaryOpFn& unary_op_fn) { in RegisterUnaryOpFn() argument
140 {op, GetPersistentStringPiece(device), type_index}, unary_op_fn)); in RegisterUnaryOpFn()
315 UnaryVariantOpRegistry::VariantUnaryOpFn* unary_op_fn = in UnaryOpVariant() local
317 if (unary_op_fn == nullptr) { in UnaryOpVariant()
323 return (*unary_op_fn)(ctx, v, v_out); in UnaryOpVariant()
423 const LocalVariantUnaryOpFn& unary_op_fn) { in UnaryVariantUnaryOpRegistration() argument
428 [type_index_name, unary_op_fn](OpKernelContext* ctx, const Variant& v, in UnaryVariantUnaryOpRegistration()
439 return unary_op_fn(ctx, t, t_out); in UnaryVariantUnaryOpRegistration()
|