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()
|