Method GetBinaryOpFn

tensorflow/core/framework/variant_op_registry.cc:217–223

Special casing BinaryOpFn per op and per device.

// Special casing BinaryOpFn per op and per device.
UnaryVariantOpRegistry::VariantBinaryOpFn*
UnaryVariantOpRegistry::GetBinaryOpFn(VariantBinaryOp op, StringPiece device,
                                      const TypeIndex& type_index) {
  auto found = binary_op_fns.find({op, device, type_index});
  if (found == binary_op_fns.end()) return nullptr;
  return &found->second;
}

void UnaryVariantOpRegistry::RegisterBinaryOpFn(
    VariantBinaryOp op, const string& device, const TypeIndex& type_index,