Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.h297 UnaryVariantOpRegistry::VariantBinaryOpFn* binary_op_fn = in BinaryOpVariants() local
299 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() argument
408 [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()