Searched refs:binary_op_fn (Results 1 – 1 of 1) sorted by relevance
297 UnaryVariantOpRegistry::VariantBinaryOpFn* binary_op_fn = in BinaryOpVariants() local299 if (binary_op_fn == nullptr) { in BinaryOpVariants()306 return (*binary_op_fn)(ctx, a, b, out); in BinaryOpVariants()404 const LocalVariantBinaryOpFn& binary_op_fn) { in UnaryVariantBinaryOpRegistration() argument408 [type_index_name, binary_op_fn](OpKernelContext* ctx, const Variant& a, in UnaryVariantBinaryOpRegistration()426 return binary_op_fn(ctx, t_a, t_b, t_out); in UnaryVariantBinaryOpRegistration()